Searched refs:shape_2d (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/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/python/keras/layers/ |
| D | local_test.py | 392 shape_2d = (int(np.prod(inputs.shape[:split_dim])),
394 inputs_2d = np.reshape(inputs, shape_2d)
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 332 TensorShape shape_2d({dim, dim}); in GrapplerItemFromMetaGraphDef() local
345 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef()
|