Searched refs:tensor_as_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.h | 219 const shape_inference::ShapeHandle& tensor_as_shape, const DataType& dtype);
|
| D | graph_properties.cc | 396 const ShapeHandle& tensor_as_shape, in MakeTensorProtoFromShape() argument
402 shape_proto->add_dim()->set_size(ic->Rank(tensor_as_shape)); in MakeTensorProtoFromShape()
405 for (int i = 0; i < ic->Rank(tensor_as_shape); i++) { in MakeTensorProtoFromShape()
406 int64 value = ic->Value(ic->Dim(tensor_as_shape, i)); in MakeTensorProtoFromShape()
434 const ShapeHandle& tensor_as_shape, in MakeConstNodeDefFromShape() argument
437 ic, MakeTensorProtoFromShape(ic, shape, tensor_as_shape, dtype), dtype); in MakeConstNodeDefFromShape()
449 const ShapeHandle& tensor_as_shape, const DataType& dtype) { in IsShapeFullyDefinedIntegerVectorOrScalar() argument
451 !ic->FullyDefined(tensor_as_shape) || in IsShapeFullyDefinedIntegerVectorOrScalar()
456 for (int32 i = 0; i < ic->Rank(tensor_as_shape); ++i) { in IsShapeFullyDefinedIntegerVectorOrScalar()
457 DimensionHandle dim = ic->Dim(tensor_as_shape, i); in IsShapeFullyDefinedIntegerVectorOrScalar()
[all …]
|