Home
last modified time | relevance | path

Searched refs:HumanString (Results 1 – 25 of 91) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc58 std::string(op_type), ShapeUtil::HumanString(shape)); in ExpectArray()
98 ShapeUtil::HumanString(accumulator_shape)); in VerifyReducerShape()
106 ShapeUtil::HumanString(accumulator_shape)); in VerifyReducerShape()
119 i, ShapeUtil::HumanString(reducer_shape.parameters(i)), in VerifyReducerShape()
120 ShapeUtil::HumanString(*accumulator_subshapes[i])); in VerifyReducerShape()
128 i, ShapeUtil::HumanString(*accumulator_subshapes[i]), in VerifyReducerShape()
129 ShapeUtil::HumanString(*init_value_shapes[i])); in VerifyReducerShape()
140 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape()
141 ShapeUtil::HumanString(input_element_shape)); in VerifyReducerShape()
154 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape()
[all …]
Dhlo_module_util.cc38 ShapeUtil::HumanString(result_shape)); in ValidateResultShape()
66 i, ShapeUtil::HumanString(program_shape.parameters(i)), in CreateModuleConfig()
67 ShapeUtil::HumanString(*argument_shapes[i])); in CreateModuleConfig()
Dtriangular_solve_expander.cc478 ShapeUtil::HumanString(a_shape), ShapeUtil::HumanString(b_shape)); in BuildTriangularSolve()
496 ShapeUtil::HumanString(a_shape), ShapeUtil::HumanString(b_shape)); in BuildTriangularSolve()
507 ShapeUtil::HumanString(a_shape)); in BuildTriangularSolve()
515 ShapeUtil::HumanString(a_shape), ShapeUtil::HumanString(b_shape)); in BuildTriangularSolve()
Dtransfer_manager.cc159 << ShapeUtil::HumanString(literal.shape()) in TransferArrayToDeviceAsync()
160 << " is not an array: " << ShapeUtil::HumanString(on_device_shape); in TransferArrayToDeviceAsync()
180 auto error = StrCat("Shape ", ShapeUtil::HumanString(shape), in TransferArrayFromDevice()
182 ShapeUtil::HumanString(HostShapeToDeviceShape(shape))); in TransferArrayFromDevice()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache_test.cc54 << "s1: " << signatures[i].HumanString() << "\n" in TEST()
55 << "s2: " << signatures[j].HumanString(); in TEST()
57 signatures[i].HumanString() == signatures[j].HumanString()) in TEST()
58 << "s1: " << signatures[i].HumanString() << "\n" in TEST()
59 << "s2: " << signatures[j].HumanString(); in TEST()
62 << "s1: " << signatures[i].HumanString() << "\n" in TEST()
64 << "s2: " << signatures[j].HumanString() << "\n" in TEST()
90 EXPECT_NE(s1.HumanString(), s2.HumanString()); in TEST()
Dxla_compilation_cache.cc193 string XlaCompilationCache::Signature::HumanString() const { in HumanString() function in tensorflow::XlaCompilationCache::Signature
241 arg.HumanString()); in BuildSignature()
528 "Try loading serialized cache entry:", sig.HumanString())); in CompileStrict()
539 VLOG(1) << "Loading cached entry for: " << sig.HumanString(); in CompileStrict()
554 "Serializing and saving cache entry: ", sig.HumanString())); in CompileStrict()
701 VLOG(3) << i << ": " << args[i].HumanString(); in CompileImpl()
738 human_signature = VLOG_IS_ON(3) ? signature.HumanString() : function.name(); in CompileImpl()
766 absl::StrAppend(out, " arg: ", arg.HumanString()); in CompileImpl()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc54 xla::ShapeUtil::HumanString(indices_shape), in XlaScatter()
56 xla::ShapeUtil::HumanString(buffer_shape), ")"); in XlaScatter()
77 xla::ShapeUtil::HumanString(buffer_shape)); in XlaScatter()
189 VLOG(3) << " Input: " << xla::ShapeUtil::HumanString(buffer_shape); in XlaScatter()
190 VLOG(3) << " Indices: " << xla::ShapeUtil::HumanString(indices_shape); in XlaScatter()
191 VLOG(3) << " Updates: " << xla::ShapeUtil::HumanString(updates_shape); in XlaScatter()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc114 xla::ShapeUtil::HumanString(then_input_shape), " vs. ", in ValidateShapes()
115 xla::ShapeUtil::HumanString(else_input_shape)); in ValidateShapes()
129 xla::ShapeUtil::HumanString(then_result.xla_output_shape), " vs. ", in ValidateShapes()
130 xla::ShapeUtil::HumanString(else_result.xla_output_shape), in ValidateShapes()
138 xla::ShapeUtil::HumanString(then_result.xla_output_shape), " vs. ", in ValidateShapes()
139 xla::ShapeUtil::HumanString(else_result.xla_output_shape)); in ValidateShapes()
159 VLOG(2) << "Input shape: " << xla::ShapeUtil::HumanString(then_input_shape); in ValidateShapes()
161 << xla::ShapeUtil::HumanString(then_result.xla_output_shape); in ValidateShapes()
214 << " shape: " << arg.HumanString() in Compile()
226 << " shape: " << arg.HumanString(); in Compile()
[all …]
Dcase_op.cc109 << " shape: " << arg.HumanString() in Compile()
122 << " shape: " << arg.HumanString(); in Compile()
226 xla::ShapeUtil::HumanString(branch0_input_shape), " vs. ", in Compile()
227 xla::ShapeUtil::HumanString(branch_input_shape))); in Compile()
231 << xla::ShapeUtil::HumanString(branch0_input_shape); in Compile()
233 << xla::ShapeUtil::HumanString( in Compile()
309 << xla::ShapeUtil::HumanString(shape_or.ValueOrDie()); in Compile()
334 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))); in Compile()
Dwhile_op.cc196 xla::ShapeUtil::HumanString(body_input_shape), " vs. ", in VerifyBodyInputAndOutputShapeMatch()
197 xla::ShapeUtil::HumanString(body_output_shape)); in VerifyBodyInputAndOutputShapeMatch()
460 VLOG(2) << "Body shape: " << xla::ShapeUtil::HumanString(body_input_shape) in Compile()
461 << " -> " << xla::ShapeUtil::HumanString(body.xla_output_shape); in Compile()
462 VLOG(2) << "Cond shape: " << xla::ShapeUtil::HumanString(cond_input_shape) in Compile()
463 << " -> " << xla::ShapeUtil::HumanString(cond.xla_output_shape); in Compile()
469 xla::ShapeUtil::HumanString(body_input_shape), " vs. ", in Compile()
470 xla::ShapeUtil::HumanString(cond_input_shape))); in Compile()
495 xla::ShapeUtil::HumanString(cond.xla_output_shape))); in Compile()
614 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))); in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression_test.cc70 TEST_F(XlaExpressionTest, HumanString) { in TEST_F() argument
71 EXPECT_EQ("invalid", XlaExpression().HumanString()); in TEST_F()
72 EXPECT_EQ("invalid", XlaExpression::Invalid().HumanString()); in TEST_F()
73 EXPECT_EQ("constant", XlaExpression::Constant(constant_).HumanString()); in TEST_F()
74 EXPECT_EQ("xla_op", XlaExpression::XlaOp(op_, DT_INT32).HumanString()); in TEST_F()
75 EXPECT_EQ("resource", XlaExpression::Resource(resource_.get()).HumanString()); in TEST_F()
Dxla_expression.cc76 string XlaExpression::HumanString() const { in HumanString() function in tensorflow::XlaExpression
110 HumanString()); in AsXlaOp()
133 HumanString()); in ResolveDynamism()
166 "ResolveConstant called on XlaExpression: ", HumanString()); in ResolveConstant()
240 << expression->HumanString(); in CastExpressionFromTensor()
250 << expression->HumanString(); in AssignExpressionToTensor()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_layout.cc28 ShapeUtil::HumanString(other_shape), in CopyLayoutFromShape()
29 ShapeUtil::HumanString(shape())); in CopyLayoutFromShape()
38 ShapeUtil::HumanString(*to_shape), in AssignLayoutToShape()
39 ShapeUtil::HumanString(shape())); in AssignLayoutToShape()
Dpacked_literal_reader.cc43 VLOG(3) << "reading shape from file: " << ShapeUtil::HumanString(shape) in Read()
78 VLOG(3) << "read shape from file: " << ShapeUtil::HumanString(shape); in Read()
Dshape.cc88 return ShapeUtil::HumanString(*this); in ToString()
249 ShapeUtil::HumanString(parameters(i))); in ToString()
252 ShapeUtil::HumanString(result())); in ToString()
Dliteral_comparison.cc342 ShapeUtil::HumanString(expected_.shape())); in Run()
562 ShapeUtil::HumanString(actual_.shape()), in ErrorMessage()
815 ShapeUtil::HumanString(actual.shape()), in NearHelper()
872 ShapeUtil::HumanString(expected), in EqualShapes()
873 ShapeUtil::HumanString(actual)); in EqualShapes()
893 ShapeUtil::HumanString(expected), in EqualShapes()
894 ShapeUtil::HumanString(actual)); in EqualShapes()
910 ShapeUtil::HumanString(expected), ShapeUtil::HumanString(actual)); in EqualShapes()
Dshape_util.cc558 CHECK(shape.IsTuple()) << HumanString(shape); in TupleElementCount()
592 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
598 DCHECK(shape.IsArray()) << ShapeUtil::HumanString(shape); in ElementsIn()
642 /* static */ std::string ShapeUtil::HumanString(const Shape& shape) { in HumanString() function in xla::ShapeUtil
654 StrAppend(&text, HumanString(elem_shape)); in HumanString()
691 std::string result = HumanString(shape); in HumanStringWithLayout()
694 std::string layout_str = LayoutUtil::HumanString(shape.layout()); in HumanStringWithLayout()
700 StrAppend(&result, LayoutUtil::HumanString(shape.layout())); in HumanStringWithLayout()
706 /* static */ std::string ShapeUtil::HumanString( in HumanString() function in xla::ShapeUtil
716 ": ", HumanString(shape))); in HumanString()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_xfeed.cc180 << ShapeUtil::HumanString(shape); in TransferLiteralToInfeedOnCpu()
190 ShapeUtil::HumanString(literal.shape())); in TransferLiteralToInfeedOnCpu()
237 << ShapeUtil::HumanString(received_shape) in TransferLiteralFromOutfeedOnCpu()
239 << ShapeUtil::HumanString(literal.shape()); in TransferLiteralFromOutfeedOnCpu()
265 << ShapeUtil::HumanString(received_shape) in TransferLiteralFromOutfeedOnCpu()
267 << ShapeUtil::HumanString(literal.shape()); in TransferLiteralFromOutfeedOnCpu()
Dxfeed_manager_test.cc49 << "want " << ShapeUtil::HumanString(shape_) << " " in Done()
51 << ShapeUtil::HumanString(shape.ValueOrDie()); in Done()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dwhile_test.cc227 << ShapeUtil::HumanString(builder.GetShape(result).value()); in XLA_TEST_F()
280 << ShapeUtil::HumanString(builder.GetShape(result).value()); in XLA_TEST_F()
339 << ShapeUtil::HumanString(builder.GetShape(result).value()); in XLA_TEST_F()
348 VLOG(2) << "expected = " << ShapeUtil::HumanString(expected.shape()); in XLA_TEST_F()
393 << ShapeUtil::HumanString(builder.GetShape(result).value()); in XLA_TEST_F()
401 VLOG(2) << "expected = " << ShapeUtil::HumanString(expected.shape()); in XLA_TEST_F()
450 << ShapeUtil::HumanString(builder.GetShape(result).value()); in XLA_TEST_F()
500 << ShapeUtil::HumanString(builder.GetShape(result).value()); in XLA_TEST_F()
506 VLOG(2) << "expected = " << ShapeUtil::HumanString(expected.shape()); in XLA_TEST_F()
547 << ShapeUtil::HumanString(builder.GetShape(result).value()); in XLA_TEST_F()
[all …]
Dtest_utils_test.cc233 << ShapeUtil::HumanString(args[0].shape()); in XLA_TEST_F()
236 << ShapeUtil::HumanString(args[1].shape()); in XLA_TEST_F()
265 << ShapeUtil::HumanString(indices_shape); in XLA_TEST_F()
303 << ShapeUtil::HumanString(indices_shape); in XLA_TEST_F()
Dliteral_test_util.cc55 LOG(INFO) << "expected: " << ShapeUtil::HumanString(expected.shape()) << " " in OnMiscompare()
57 LOG(INFO) << "actual: " << ShapeUtil::HumanString(actual.shape()) << " " in OnMiscompare()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dbroadcast.cc40 "Input shape (", ShapeUtil::HumanString(input_shape), in BroadcastTo()
55 ShapeUtil::HumanString(input_shape), " to [", in BroadcastTo()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddumped_computation_to_operation_list.cc50 absl::StrAppend(out, ShapeUtil::HumanString(operand->shape())); in DefaultAction()
55 ShapeUtil::HumanString(hlo->shape()), path_); in DefaultAction()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc267 << ShapeUtil::HumanString(old_shape) << " to " in TryResolvePaddedShapesForTensorCore()
268 << ShapeUtil::HumanString(new_shape) << ", a size increase of " in TryResolvePaddedShapesForTensorCore()
437 << ShapeUtil::HumanString(old_shape) << " to " in TryResolvePaddedShapesForIntegerConvolution()
438 << ShapeUtil::HumanString(new_shape) << ", a size increase of " in TryResolvePaddedShapesForIntegerConvolution()

1234