Home
last modified time | relevance | path

Searched refs:shape_1d (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder_test.cc279 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()
Dgrappler_item_builder.cc232 TensorShape shape_1d({dim}); in GrapplerItemFromMetaGraphDef() local
238 Tensor value_tensor(input.dtype(), shape_1d); in GrapplerItemFromMetaGraphDef()
255 Tensor dense_shape_tensor(DT_INT64, shape_1d); in GrapplerItemFromMetaGraphDef()