Searched refs:ParseShape (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | while_loop_simplifier_test.cc | 557 ParseShape("(s32[1], s32[2], s32[3], s32[4])").ValueOrDie(); in TEST_F() 569 ParseShape("((s32[1]), (s32[2], s32[3], (s32[4])))").ValueOrDie())); in TEST_F() 642 Shape new_while_shape = ParseShape("(s32[1], s32[3])").ValueOrDie(); in TEST_F() 654 ParseShape("(s32[1], s32[2], s32[3])").ValueOrDie())); in TEST_F() 712 ParseShape("(s32[], s32[], s32[], s32[])").ValueOrDie(); in TEST_F()
|
D | hlo_parser.h | 69 StatusOr<Shape> ParseShape(absl::string_view str);
|
D | hlo_matchers.h | 358 new ::xla::testing::HloShapeMatcher(ParseShape(shape).ValueOrDie())); in Shape() 368 ParseShape(shape).ValueOrDie())); in ShapeWithLayout()
|
D | hlo_parser.cc | 299 bool ParseShape(Shape* result); 662 if (!ParseShape(&shape) || !ParseOpcode(&opcode)) { in ParseInstructionRhs() 2545 if (!ParseShape(&shape.value())) { in ParseOperands() 3303 if (!ParseShape(&shape)) { in ParseShapeList() 3357 return ParseShape(shape); in ParseParamListToShape() 3381 if (!ParseName(&name) || !ParseShape(&shape)) { in ParseParamList() 3524 bool HloParser::ParseShape(Shape* result) { in ParseShape() function in xla::__anonc071bf1f0111::HloParser 3533 if (!ParseShape(&shapes.back())) { in ParseShape() 3974 if (!ParseShape(&shape)) { in ParseShapeOnly() 4146 StatusOr<Shape> ParseShape(absl::string_view str) { in ParseShape() function
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | text_literal_reader.cc | 70 TF_ASSIGN_OR_RETURN(Shape shape, ParseShape(shape_string)); in ReadAllLines()
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | replay_computation.cc | 149 xfeed_shape = std::move(ParseShape(fake_xfeed_shape)).ValueOrDie(); in GetXfeedShape()
|