Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dabstract_tensor_handle.cc23 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()
H A Dimmediate_execution_tensor_handle.cc22 std::string shape_string; in DebugString() local
24 shape_string = shape.DebugString(); in DebugString()
26 shape_string = "<error computing shape>"; in DebugString()
42 return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string, in DebugString()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dtext_literal_reader.cc62 std::string shape_string; in ReadAllLines() local
63 Status s = buf.ReadLine(&shape_string); in ReadAllLines()
68 absl::StripAsciiWhitespace(&shape_string); in ReadAllLines()
69 TF_ASSIGN_OR_RETURN(Shape shape, ParseShape(shape_string)); in ReadAllLines()
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dstrip_unused_nodes.cc87 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()
H A Dtransform_utils.cc597 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()
H A Dtransform_utils.h137 Status TensorShapeFromString(const string& shape_string, TensorShape* result);
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/
H A Dhlo_proto_to_memory_visualization_utils.cc101 const std::string& shape_string, in MakeHeapObject() argument
114 result.set_shape_string(shape_string); in MakeHeapObject()
421 std::string shape_string = ShapeUtil::HumanStringWithLayout(*shape); in ConvertHloProtoToPreprocessResult() local
425 std::string label = StrFormat("%s: %s # %s", instruction_name, shape_string, in ConvertHloProtoToPreprocessResult()
428 hlo_instruction->metadata().op_name(), shape_string, in ConvertHloProtoToPreprocessResult()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc2876 auto shape_string = [&]() { in InferWhileShape() local
2886 shape_string()); in InferWhileShape()
2894 shape_string()); in InferWhileShape()
2924 auto shape_string = [&]() { in InferConditionalShape() local
2932 j, j, shape_string()); in InferConditionalShape()
2937 auto shape_string = [&]() { in InferConditionalShape() local
2946 j, shape_string()); in InferConditionalShape()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/protobuf/
H A Dmemory_viewer_preprocess.proto21 string shape_string = 8; field