| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | split_test.cc | 275 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
287 input_shape, input_data, axis_shape, axis_data, output1_shape, golden1, in TF_LITE_MICRO_TEST()
295 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
307 input_shape, input_data, axis_shape, axis_data, output1_shape, golden1, in TF_LITE_MICRO_TEST()
315 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
327 input_shape, input_data, axis_shape, axis_data, output1_shape, golden1, in TF_LITE_MICRO_TEST()
335 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
347 input_shape, input_data, axis_shape, axis_data, output1_shape, golden1, in TF_LITE_MICRO_TEST()
355 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
367 input_shape, input_data, axis_shape, axis_data, output1_shape, golden1, in TF_LITE_MICRO_TEST()
[all …]
|
| D | split_v_test.cc | 115 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
139 tflite::testing::TestSplitVFloat(input_shape, input_values, axis_shape, in TF_LITE_MICRO_TEST()
153 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
173 tflite::testing::TestSplitVFloat(input_shape, input_values, axis_shape, in TF_LITE_MICRO_TEST()
187 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
207 tflite::testing::TestSplitVFloat(input_shape, input_values, axis_shape, in TF_LITE_MICRO_TEST()
221 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
241 tflite::testing::TestSplitVFloat(input_shape, input_values, axis_shape, in TF_LITE_MICRO_TEST()
254 int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
274 tflite::testing::TestSplitVFloat(input_shape, input_values, axis_shape, in TF_LITE_MICRO_TEST()
[all …]
|
| D | reduce_test.cc | 484 const int axis_shape[] = {1, 4}; in TF_LITE_MICRO_TEST() local
493 input_shape, input_data, axis_shape, axis_data, output_shape, output_data, in TF_LITE_MICRO_TEST()
502 const int axis_shape[] = {1, 2}; in TF_LITE_MICRO_TEST() local
511 input_shape, input_data, axis_shape, axis_data, output_shape, output_data, in TF_LITE_MICRO_TEST()
518 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
534 axis_shape, axis_data, output_shape, expected_output_data, in TF_LITE_MICRO_TEST()
542 const int axis_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
560 axis_shape, axis_data, output_shape, expected_output_data, in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | roll_op.cc | 32 const TensorShape axis_shape = ctx->InputShape(2); in Compile() local
41 OP_REQUIRES(ctx, axis_shape.dims() <= 1, in Compile()
46 ctx, shift_shape == axis_shape, in Compile()
54 int64 num_axes = axis_shape.dims() == 0 ? 1 : axis_shape.dim_size(0); in Compile()
56 int64 cur_axis = axis_shape.dims() == 0 ? *axis.GetIntegralAsS64({}) in Compile()
|
| D | quantize_and_dequantize_op.cc | 162 xla::Shape axis_shape = b->GetShape(min_range).ValueOrDie(); in Compile() local
165 if (!xla::ShapeUtil::IsScalar(axis_shape)) { in Compile()
|
| D | gather_op.cc | 164 const TensorShape axis_shape = context->InputShape(2); in XlaGatherWithBatchDimsOpImpl() local
165 if (!TensorShapeUtils::IsScalar(axis_shape)) { in XlaGatherWithBatchDimsOpImpl()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | reduce_test.cc | 26 const TensorData& output, std::initializer_list<int> axis_shape, in ReduceOpModel() argument
29 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in ReduceOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 83 std::initializer_list<int> axis_shape, in MeanOpConstModel() argument
86 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MeanOpConstModel()
112 std::initializer_list<int> axis_shape, in SumOpConstModel() argument
115 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in SumOpConstModel()
141 std::initializer_list<int> axis_shape, in ProdOpConstModel() argument
144 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in ProdOpConstModel()
170 std::initializer_list<int> axis_shape, in MaxOpConstModel() argument
173 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MaxOpConstModel()
199 std::initializer_list<int> axis_shape, in MinOpConstModel() argument
202 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MinOpConstModel()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 4714 std::initializer_list<int> axis_shape, in MeanOpConstModel() argument
4717 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MeanOpConstModel()
|