Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc357 const ShapeHandle& tensor_as_shape, const DataType& dtype) { in IsShapeFullyDefinedIntegerVectorOrScalar() argument
359 !ic->FullyDefined(tensor_as_shape) || in IsShapeFullyDefinedIntegerVectorOrScalar()
370 const ShapeHandle& tensor_as_shape, in MakeTensorProtoFromShape() argument
376 shape_proto->add_dim()->set_size(ic->Rank(tensor_as_shape)); in MakeTensorProtoFromShape()
379 for (int i = 0; i < ic->Rank(tensor_as_shape); i++) { in MakeTensorProtoFromShape()
380 int64 value = ic->Value(ic->Dim(tensor_as_shape, i)); in MakeTensorProtoFromShape()
408 const ShapeHandle& tensor_as_shape, in MakeConstNodeDefFromShape() argument
411 ic, MakeTensorProtoFromShape(ic, shape, tensor_as_shape, dtype), dtype); in MakeConstNodeDefFromShape()