Searched refs:shape_string (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | abstract_tensor_handle.cc | 23 std::string shape_string; in DebugString() local
25 shape_string = "<error computing shape>"; in DebugString()
27 shape_string = shape.DebugString(); in DebugString()
29 return absl::StrCat("TensorHandle(shape=", shape_string, in DebugString()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_rewriter_transform.cc | 54 string shape_string; in RewriteQuantizedStrippedModelForHexagon() local
56 INPUT_SHAPE_PREFIX + std::to_string(i), "", &shape_string)); in RewriteQuantizedStrippedModelForHexagon()
57 std::vector<string> split_shape = str_util::Split(shape_string, ','); in RewriteQuantizedStrippedModelForHexagon()
74 << ", shape = " << shape_string in RewriteQuantizedStrippedModelForHexagon()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | text_literal_reader.cc | 63 string shape_string; in ReadAllLines() local
64 Status s = buf.ReadLine(&shape_string); in ReadAllLines()
69 absl::StripAsciiWhitespace(&shape_string); in ReadAllLines()
70 TF_ASSIGN_OR_RETURN(Shape shape, ParseShape(shape_string)); in ReadAllLines()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | strip_unused_nodes.cc | 87 const string& shape_string = context.params.at("shape")[0]; in ShapeForPlaceholder() local
88 TF_RETURN_IF_ERROR(TensorShapeFromString(shape_string, result)); in ShapeForPlaceholder()
105 const string& shape_string = context.params.at("shape_for_name")[i]; in ShapeForPlaceholder() local
106 TF_RETURN_IF_ERROR(TensorShapeFromString(shape_string, result)); in ShapeForPlaceholder()
|
| D | transform_utils.cc | 597 Status TensorShapeFromString(const string& shape_string, TensorShape* result) { in TensorShapeFromString() argument
598 if (shape_string.empty()) { in TensorShapeFromString()
601 std::vector<string> dims_as_str = str_util::Split(shape_string, ","); in TensorShapeFromString()
608 return errors::InvalidArgument("Could parse as shape: '", shape_string, in TensorShapeFromString()
|
| D | transform_utils.h | 137 Status TensorShapeFromString(const string& shape_string, TensorShape* result);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cse.cc | 69 string shape_string = shape.ShortDebugString(); in CombineConstants() local
72 auto range = constants.equal_range(shape_string); in CombineConstants()
82 constants.emplace(shape_string, instruction); in CombineConstants()
|
| D | shape_inference.cc | 2759 auto shape_string = [&]() { in InferWhileShape() local
2769 shape_string()); in InferWhileShape()
2777 shape_string()); in InferWhileShape()
2807 auto shape_string = [&]() { in InferConditionalShape() local
2815 j, j, shape_string()); in InferConditionalShape()
2820 auto shape_string = [&]() { in InferConditionalShape() local
2829 j, shape_string()); in InferConditionalShape()
|