Searched refs:shape_1d (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | softmax_test.cc | 42 const int shape_1d[] = {1, 5}; variable
335 tflite::testing ::shape_1d, tflite::testing::input_data_1d, in TF_LITE_MICRO_TEST()
336 tflite::testing::shape_1d, tflite::testing::golden_1d, output_data); in TF_LITE_MICRO_TEST()
347 tflite::testing::shape_1d, tflite::testing::input_data_1d, in TF_LITE_MICRO_TEST()
348 input_quantized, input_scale, input_zero_point, tflite::testing::shape_1d, in TF_LITE_MICRO_TEST()
362 tflite::testing::shape_1d, tflite::testing::input_data_1d, in TF_LITE_MICRO_TEST()
363 input_quantized, input_scale, input_zero_point, tflite::testing::shape_1d, in TF_LITE_MICRO_TEST()
377 tflite::testing::shape_1d, tflite::testing::input_data_1d, in TF_LITE_MICRO_TEST()
378 input_quantized, input_scale, input_zero_point, tflite::testing::shape_1d, in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder_test.cc | 279 auto shape_1d = PartialTensorShape({-1}); in TEST_F() local
281 ops::Placeholder::Shape(shape_1d)); in TEST_F()
291 shape_1d.AsProto(input.mutable_tensor_shape()); in TEST_F()
|
| D | grappler_item_builder.cc | 332 TensorShape shape_1d({dim}); in GrapplerItemFromMetaGraphDef() local
338 Tensor value_tensor(input.dtype(), shape_1d); in GrapplerItemFromMetaGraphDef()
355 Tensor dense_shape_tensor(DT_INT64, shape_1d); in GrapplerItemFromMetaGraphDef()
|