Searched defs:axis_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | split_v_test.cc | 115 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
153 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
187 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
221 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
254 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
288 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
333 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
410 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
|
| D | split_test.cc | 275 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
295 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
315 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
335 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
355 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
374 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
402 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
422 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
442 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
|
| D | reduce_test.cc | 484 const int axis_shape[] = {1, 4}; in TF_LITE_MICRO_TEST() local
502 const int axis_shape[] = {1, 2}; in TF_LITE_MICRO_TEST() local
518 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
542 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | roll_op.cc | 32 const TensorShape axis_shape = ctx->InputShape(2); in Compile() local
|
| D | quantize_and_dequantize_op.cc | 162 xla::Shape axis_shape = b->GetShape(min_range).ValueOrDie(); in Compile() local
|
| D | gather_op.cc | 164 const TensorShape axis_shape = context->InputShape(2); in XlaGatherWithBatchDimsOpImpl() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | reduce_test.cc | 26 const TensorData& output, std::initializer_list<int> axis_shape, in ReduceOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 83 std::initializer_list<int> axis_shape, in MeanOpConstModel()
112 std::initializer_list<int> axis_shape, in SumOpConstModel()
141 std::initializer_list<int> axis_shape, in ProdOpConstModel()
170 std::initializer_list<int> axis_shape, in MaxOpConstModel()
199 std::initializer_list<int> axis_shape, in MinOpConstModel()
228 std::initializer_list<int> axis_shape, in AnyOpConstModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 4714 std::initializer_list<int> axis_shape, in MeanOpConstModel()
|