Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_parser_test.cc3676 std::string shape_string = "f32[123,456]"; local
3685 std::string shape_string = "(f32[1572864],s8[5120,1024])"; local
3696 std::string shape_string = "(f32[1],(f32[2], token[]), opaque[], f32[3])"; local
3711 std::string shape_string = "f32[123,456]{0,1}"; local
3721 std::string shape_string = "f32[123,456]{0,1:T(2,128)}"; local
3776 std::string shape_string = "pred[123,456]{1,0:T(2,128)E(1)S(3)}"; local
3822 for (const std::string& shape_string : shape_strings) { local
3829 std::string shape_string = "f32[123,<=456]"; local
3838 std::string shape_string = "(f32[42], u32[<=123,<=456])"; local
H A Dshape_inference.cc2876 auto shape_string = [&]() { in InferWhileShape() local
2924 auto shape_string = [&]() { in InferConditionalShape() local
2937 auto shape_string = [&]() { in InferConditionalShape() local
/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
105 const string& shape_string = context.params.at("shape_for_name")[i]; in ShapeForPlaceholder() local
H A Dtransform_utils.cc597 Status TensorShapeFromString(const string& shape_string, TensorShape* result) { in TensorShapeFromString()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dabstract_tensor_handle.cc23 std::string shape_string; in DebugString() local
H A Dimmediate_execution_tensor_handle.cc22 std::string shape_string; in DebugString() local
/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()
421 std::string shape_string = ShapeUtil::HumanStringWithLayout(*shape); in ConvertHloProtoToPreprocessResult() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dtext_literal_reader.cc62 std::string shape_string; in ReadAllLines() local
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/protobuf/
H A Dmemory_viewer_preprocess.proto21 string shape_string = 8; field