Searched defs:scratch_shape (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | LSTMLayerFixture.h | 51 …TensorShape scratch_shape, ActivationLayerInfo info, float cell_threshold, float projection_thresh… in setup()
80 …rShape &output_cell_shape, const TensorShape &output_shape, const TensorShape &scratch_shape, Acti… in compute_target()
325 …rShape &output_cell_shape, const TensorShape &output_shape, const TensorShape &scratch_shape, Acti… in compute_reference()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n.cc | 83 TfLiteIntArray* scratch_shape = TfLiteIntArrayCreate(1); in Prepare() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1102 const RuntimeShape& scratch_shape, int32_t* scratch, int32_t* row_sums, in HybridConvPerChannel()
|