Searched refs:shape_2d (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | utils_test.cc | 46 TfLiteIntArray* shape_2d = TfLiteIntArrayCreate(2); in TEST() local
47 shape_2d->data[0] = 4; in TEST()
48 shape_2d->data[1] = 3; in TEST()
50 Get4DShape(&batch_dim, &height_dim, &width_dim, &depth_dim, shape_2d), in TEST()
54 EXPECT_EQ(width_dim, shape_2d->data[0]); in TEST()
55 EXPECT_EQ(depth_dim, shape_2d->data[1]); in TEST()
57 TfLiteIntArrayFree(shape_2d); in TEST()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | softmax_test.cc | 50 const int shape_2d[] = {2, 2, 5}; variable
387 tflite::testing ::shape_2d, tflite::testing::input_data_2d, in TF_LITE_MICRO_TEST()
388 tflite::testing::shape_2d, tflite::testing::golden_2d, output_data); in TF_LITE_MICRO_TEST()
399 tflite::testing::shape_2d, tflite::testing::input_data_2d, in TF_LITE_MICRO_TEST()
400 input_quantized, input_scale, input_zero_point, tflite::testing::shape_2d, in TF_LITE_MICRO_TEST()
414 tflite::testing::shape_2d, tflite::testing::input_data_2d, in TF_LITE_MICRO_TEST()
415 input_quantized, input_scale, input_zero_point, tflite::testing::shape_2d, in TF_LITE_MICRO_TEST()
429 tflite::testing::shape_2d, tflite::testing::input_data_2d, in TF_LITE_MICRO_TEST()
430 input_quantized, input_scale, input_zero_point, tflite::testing::shape_2d, in TF_LITE_MICRO_TEST()
553 tflite::testing::shape_2d, tflite::testing::input_data_2d, in TF_LITE_MICRO_TEST()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local_test.py | 396 shape_2d = (int(np.prod(inputs.shape[:split_dim])),
398 inputs_2d = np.reshape(inputs, shape_2d)
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 333 TensorShape shape_2d({dim, dim}); in GrapplerItemFromMetaGraphDef() local
346 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef()
|