Searched refs:shape_string (Results 1 – 7 of 7) sorted by relevance
| /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()
58 CHECK(str_util::SplitAndParseAsInts(shape_string, ',', &dims)); in RewriteQuantizedStrippedModelForHexagon()
69 << ", 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 | 88 const string& shape_string = context.params.at("shape")[0]; in ShapeForPlaceholder() local
89 TF_RETURN_IF_ERROR(TensorShapeFromString(shape_string, result)); in ShapeForPlaceholder()
106 const string& shape_string = context.params.at("shape_for_name")[i]; in ShapeForPlaceholder() local
107 TF_RETURN_IF_ERROR(TensorShapeFromString(shape_string, result)); in ShapeForPlaceholder()
|
| D | transform_utils.cc | 595 Status TensorShapeFromString(const string& shape_string, TensorShape* result) { in TensorShapeFromString() argument
596 if (shape_string.empty()) { in TensorShapeFromString()
600 if (!str_util::SplitAndParseAsInts(shape_string, ',', &dims)) { in TensorShapeFromString()
601 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 | 68 string shape_string = shape.ShortDebugString(); in CombineConstants() local
71 auto range = constants.equal_range(shape_string); in CombineConstants()
81 constants.emplace(shape_string, instruction); in CombineConstants()
|
| D | shape_inference.cc | 2532 auto shape_string = [&]() { in InferWhileShape() local
2541 shape_string()); in InferWhileShape()
2549 shape_string()); in InferWhileShape()
2579 auto shape_string = [&]() { in InferConditionalShape() local
2587 j, j, shape_string()); in InferConditionalShape()
2592 auto shape_string = [&]() { in InferConditionalShape() local
2601 j, shape_string()); in InferConditionalShape()
|