Searched refs:ParseShape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier_test.cc | 611 ParseShape("(s32[1], s32[2], s32[3], s32[4])").ValueOrDie(); in TEST_F()
623 ParseShape("((s32[1]), (s32[2], s32[3], (s32[4])))").ValueOrDie())); in TEST_F()
696 Shape new_while_shape = ParseShape("(s32[1], s32[3])").ValueOrDie(); in TEST_F()
708 ParseShape("(s32[1], s32[2], s32[3])").ValueOrDie())); in TEST_F()
766 ParseShape("(s32[], s32[], s32[], s32[])").ValueOrDie(); in TEST_F()
831 Shape new_while_shape = ParseShape("(s32[], s32[])").ValueOrDie(); in TEST_F()
|
| D | hlo_parser.h | 78 StatusOr<Shape> ParseShape(absl::string_view str);
|
| D | hlo_matchers.h | 389 new ::xla::testing::HloShapeMatcher(ParseShape(shape).ValueOrDie())); in Shape()
399 ParseShape(shape).ValueOrDie())); in ShapeWithLayout()
|
| D | hlo_parser.cc | 449 bool ParseShape(Shape* result);
988 if ((parse_shape && !ParseShape(&shape)) || !ParseOpcode(&opcode)) { in ParseInstructionRhs()
3061 if (!ParseShape(&literal_shape)) { in ParseLiteral()
3438 if (!ParseShape(&shape.value())) { in ParseOperands()
3804 if (!ParseShape(&result)) { in ParseAttributeHelper()
4303 if (!ParseShape(&shape)) { in ParseShapeList()
4379 return ParseShape(shape); in ParseParamListToShape()
4403 if (!ParseName(&name) || !ParseShape(&shape)) { in ParseParamList()
4567 bool HloParserImpl::ParseShape(Shape* result) { in ParseShape() function in xla::__anonf806807d0111::HloParserImpl
4576 if (!ParseShape(&shapes.back())) { in ParseShape()
[all …]
|
| /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/lite/micro/ |
| D | micro_mutable_op_resolver.h | 378 return AddBuiltin(BuiltinOperator_SHAPE, Register_SHAPE(), ParseShape); in AddShape()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.h | 282 TfLiteStatus ParseShape(const Operator* op, ErrorReporter* error_reporter,
|
| D | flatbuffer_conversions.cc | 410 return ParseShape(op, error_reporter, allocator, builtin_data); in ParseOpDataTfLite()
1753 TfLiteStatus ParseShape(const Operator* op, ErrorReporter* error_reporter, in ParseShape() function
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 174 xfeed_shape = std::move(ParseShape(fake_xfeed_shape)).ValueOrDie(); in GetXfeedShape()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 165 return absl::make_unique<Shape>(ValueOrThrow(ParseShape(s))); in BuildXlaCompilerSubmodule()
|