HomeSort by relevance Sort by last modified time
    Searched refs:scratchShape (Results 1 - 3 of 3) sorted by null

  /frameworks/ml/nn/common/operations/
LSTM.cpp 232 Shape *scratchShape,
290 scratchShape->dimensions = { n_batch, n_cell * 3 };
293 scratchShape->dimensions = { n_batch, n_cell * 4 };
295 scratchShape->type = inputShape.type;
296 scratchShape->offset = inputShape.offset;
297 scratchShape->scale = inputShape.scale;
LSTM.h 55 Shape *scratchShape,
  /frameworks/ml/nn/common/
CpuExecutor.cpp     [all...]

Completed in 73 milliseconds