Searched refs:tensor_as_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.h | 220 const shape_inference::ShapeHandle& tensor_as_shape, const DataType& dtype);
|
| D | graph_properties.cc | 400 const ShapeHandle& tensor_as_shape, in MakeTensorProtoFromShape() argument
406 shape_proto->add_dim()->set_size(ic->Rank(tensor_as_shape)); in MakeTensorProtoFromShape()
409 for (int i = 0; i < ic->Rank(tensor_as_shape); i++) { in MakeTensorProtoFromShape()
410 int64_t value = ic->Value(ic->Dim(tensor_as_shape, i)); in MakeTensorProtoFromShape()
438 const ShapeHandle& tensor_as_shape, in MakeConstNodeDefFromShape() argument
441 ic, MakeTensorProtoFromShape(ic, shape, tensor_as_shape, dtype), dtype); in MakeConstNodeDefFromShape()
500 const ShapeHandle& tensor_as_shape, const DataType& dtype) { in IsShapeFullyDefinedIntegerVectorOrScalar() argument
502 !ic->FullyDefined(tensor_as_shape) || in IsShapeFullyDefinedIntegerVectorOrScalar()
507 for (int32_t i = 0; i < ic->Rank(tensor_as_shape); ++i) { in IsShapeFullyDefinedIntegerVectorOrScalar()
508 DimensionHandle dim = ic->Dim(tensor_as_shape, i); in IsShapeFullyDefinedIntegerVectorOrScalar()
[all …]
|