Searched refs:shape_2d (Results 1 – 2 of 2) sorted by relevance
46 TfLiteIntArray* shape_2d = TfLiteIntArrayCreate(2); in TEST() local47 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()
334 TensorShape shape_2d({dim, dim}); in GrapplerItemFromMetaGraphDef() local347 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef()