Searched refs:shape_1d (Results 1 – 2 of 2) sorted by relevance
279 auto shape_1d = PartialTensorShape({-1}); in TEST_F() local281 ops::Placeholder::Shape(shape_1d)); in TEST_F()291 shape_1d.AsProto(input.mutable_tensor_shape()); in TEST_F()
232 TensorShape shape_1d({dim}); in GrapplerItemFromMetaGraphDef() local238 Tensor value_tensor(input.dtype(), shape_1d); in GrapplerItemFromMetaGraphDef()255 Tensor dense_shape_tensor(DT_INT64, shape_1d); in GrapplerItemFromMetaGraphDef()