/external/tensorflow/tensorflow/compiler/xla/service/ |
D | shape_inference.cc | 55 string(op_type), ShapeUtil::HumanString(shape)); in ExpectArray() 95 ShapeUtil::HumanString(accumulator_shape)); in VerifyReducerShape() 103 ShapeUtil::HumanString(accumulator_shape)); in VerifyReducerShape() 116 i, ShapeUtil::HumanString(reducer_shape.parameters(i)), in VerifyReducerShape() 117 ShapeUtil::HumanString(*accumulator_subshapes[i])); in VerifyReducerShape() 125 i, ShapeUtil::HumanString(*accumulator_subshapes[i]), in VerifyReducerShape() 126 ShapeUtil::HumanString(*init_value_shapes[i])); in VerifyReducerShape() 137 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape() 138 ShapeUtil::HumanString(input_element_shape)); in VerifyReducerShape() 151 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), in VerifyReducerShape() [all …]
|
D | triangular_solve_expander.cc | 362 ShapeUtil::HumanString(a_shape), ShapeUtil::HumanString(b_shape)); in BuildTriangularSolve() 379 ShapeUtil::HumanString(a_shape), ShapeUtil::HumanString(b_shape)); in BuildTriangularSolve() 389 ShapeUtil::HumanString(a_shape)); in BuildTriangularSolve() 397 ShapeUtil::HumanString(a_shape), ShapeUtil::HumanString(b_shape)); in BuildTriangularSolve()
|
D | transfer_manager.cc | 161 << ShapeUtil::HumanString(literal.shape()) in TransferArrayToDeviceAsync() 162 << " is not an array: " << ShapeUtil::HumanString(on_device_shape); in TransferArrayToDeviceAsync() 182 auto error = StrCat("Shape ", ShapeUtil::HumanString(shape), in TransferArrayFromDevice() 184 ShapeUtil::HumanString(HostShapeToDeviceShape(shape))); in TransferArrayFromDevice()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | if_op.cc | 83 << " shape: " << arg.HumanString() in Compile() 92 << " shape: " << arg.HumanString(); in Compile() 164 xla::ShapeUtil::HumanString(then_input_shape), " vs. ", in Compile() 165 xla::ShapeUtil::HumanString(else_input_shape))); in Compile() 174 xla::ShapeUtil::HumanString(then_result.xla_output_shape), " vs. ", in Compile() 175 xla::ShapeUtil::HumanString(else_result.xla_output_shape))); in Compile() 177 VLOG(2) << "Input shape: " << xla::ShapeUtil::HumanString(then_input_shape); in Compile() 179 << xla::ShapeUtil::HumanString(then_result.xla_output_shape); in Compile() 234 << xla::ShapeUtil::HumanString(shape_or.ValueOrDie()); in Compile() 254 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))); in Compile()
|
D | case_op.cc | 80 << " shape: " << arg.HumanString() in Compile() 89 << " shape: " << arg.HumanString(); in Compile() 168 xla::ShapeUtil::HumanString(branch0_input_shape), " vs. ", in Compile() 169 xla::ShapeUtil::HumanString(branch_input_shape))); in Compile() 178 xla::ShapeUtil::HumanString(branch_results[0].xla_output_shape), in Compile() 180 xla::ShapeUtil::HumanString(branch_results[j].xla_output_shape))); in Compile() 184 << xla::ShapeUtil::HumanString(branch0_input_shape); in Compile() 186 << xla::ShapeUtil::HumanString( in Compile() 246 << xla::ShapeUtil::HumanString(shape_or.ValueOrDie()); in Compile() 266 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))); in Compile()
|
D | xla_select_and_scatter_op.cc | 85 xla::ShapeUtil::HumanString(select_output_shape), " got ", in Compile() 86 xla::ShapeUtil::HumanString(select.xla_output_shape))); in Compile() 107 xla::ShapeUtil::HumanString(scalar_shape), " got ", in Compile() 108 xla::ShapeUtil::HumanString(scatter.xla_output_shape))); in Compile()
|
D | while_op.cc | 173 xla::ShapeUtil::HumanString(body_input_shape), " vs. ", in VerifyBodyInputAndOutputShapeMatch() 174 xla::ShapeUtil::HumanString(body_output_shape)); in VerifyBodyInputAndOutputShapeMatch() 428 VLOG(2) << "Body shape: " << xla::ShapeUtil::HumanString(body_input_shape) in Compile() 429 << " -> " << xla::ShapeUtil::HumanString(body.xla_output_shape); in Compile() 430 VLOG(2) << "Cond shape: " << xla::ShapeUtil::HumanString(cond_input_shape) in Compile() 431 << " -> " << xla::ShapeUtil::HumanString(cond.xla_output_shape); in Compile() 437 xla::ShapeUtil::HumanString(body_input_shape), " vs. ", in Compile() 438 xla::ShapeUtil::HumanString(cond_input_shape))); in Compile() 463 xla::ShapeUtil::HumanString(cond.xla_output_shape))); in Compile() 548 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))); in Compile()
|
D | xla_reduce_op.cc | 83 xla::ShapeUtil::HumanString(scalar_shape), " got ", in Compile() 84 xla::ShapeUtil::HumanString(reducer.xla_output_shape))); in Compile()
|
/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
D | scatter.cc | 54 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()
|
D | broadcast.cc | 41 "Input shape (", xla::ShapeUtil::HumanString(input_shape), in BroadcastTo() 55 xla::ShapeUtil::HumanString(input_shape), in BroadcastTo()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_expression_test.cc | 70 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()
|
D | xla_expression.cc | 65 string XlaExpression::HumanString() const { in HumanString() function in tensorflow::XlaExpression 99 HumanString()); in AsXlaOp() 117 "ResolveConstant called on XlaExpression: ", HumanString()); in ResolveConstant()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | shape_layout.cc | 28 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()
|
D | packed_literal_reader.cc | 44 VLOG(3) << "reading shape from file: " << ShapeUtil::HumanString(shape) in Read() 79 VLOG(3) << "read shape from file: " << ShapeUtil::HumanString(shape); in Read()
|
D | shape.cc | 82 return ShapeUtil::HumanString(*this); in ToString() 213 ShapeUtil::HumanString(parameters(i))); in ToString() 216 ShapeUtil::HumanString(result())); in ToString()
|
D | shape_util.cc | 339 CHECK(shape.IsTuple()) << HumanString(shape); in TupleElementCount() 373 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape() 379 DCHECK(shape.IsArray()) << ShapeUtil::HumanString(shape); in ElementsIn() 423 /* static */ string ShapeUtil::HumanString(const Shape& shape) { in HumanString() function in xla::ShapeUtil 428 StrAppend(&text, prefix, HumanString(elem_shape)); in HumanString() 466 string layout_str = LayoutUtil::HumanString(shape.layout()); in HumanStringWithLayout() 472 StrAppend(&result, LayoutUtil::HumanString(shape.layout())); in HumanStringWithLayout() 477 /* static */ string ShapeUtil::HumanString(const ProgramShape& program_shape) { in HumanString() function in xla::ShapeUtil 484 ": ", HumanString(shape))); in HumanString() 487 HumanString(program_shape.result())); in HumanString() [all …]
|
D | literal_comparison.cc | 322 ShapeUtil::HumanString(expected_.shape())); in Run() 556 ShapeUtil::HumanString(actual_.shape()), in ErrorMessage() 772 ShapeUtil::HumanString(actual.shape()), in NearHelper() 823 ShapeUtil::HumanString(expected), in EqualShapes() 824 ShapeUtil::HumanString(actual)); in EqualShapes() 844 ShapeUtil::HumanString(expected), in EqualShapes() 845 ShapeUtil::HumanString(actual)); in EqualShapes() 861 ShapeUtil::HumanString(expected), ShapeUtil::HumanString(actual)); in EqualShapes()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_transfer_manager.cc | 97 << ShapeUtil::HumanString(shape); in TransferLiteralToInfeed() 107 ShapeUtil::HumanString(literal.shape())); in TransferLiteralToInfeed() 192 << ShapeUtil::HumanString(received_shape) in TransferLiteralFromOutfeed() 194 << ShapeUtil::HumanString(literal_shape); in TransferLiteralFromOutfeed() 218 << ShapeUtil::HumanString(received_shape) in TransferLiteralFromOutfeed() 220 << ShapeUtil::HumanString(literal_shape); in TransferLiteralFromOutfeed()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | while_test.cc | 228 << ShapeUtil::HumanString( in XLA_TEST_F() 282 << ShapeUtil::HumanString( in XLA_TEST_F() 342 << ShapeUtil::HumanString( in XLA_TEST_F() 352 VLOG(2) << "expected = " << ShapeUtil::HumanString(expected.shape()); in XLA_TEST_F() 397 << ShapeUtil::HumanString( in XLA_TEST_F() 406 VLOG(2) << "expected = " << ShapeUtil::HumanString(expected.shape()); in XLA_TEST_F() 455 << ShapeUtil::HumanString( in XLA_TEST_F() 506 << ShapeUtil::HumanString( in XLA_TEST_F() 513 VLOG(2) << "expected = " << ShapeUtil::HumanString(expected.shape()); in XLA_TEST_F() 553 << ShapeUtil::HumanString( in XLA_TEST_F() [all …]
|
D | test_utils_test.cc | 251 << ShapeUtil::HumanString(args[0].shape()); in XLA_TEST_F() 254 << ShapeUtil::HumanString(args[1].shape()); in XLA_TEST_F() 283 << ShapeUtil::HumanString(indices_shape); in XLA_TEST_F() 321 << ShapeUtil::HumanString(indices_shape); in XLA_TEST_F()
|
D | literal_test_util.cc | 48 LOG(INFO) << "expected: " << ShapeUtil::HumanString(expected.shape()) << " " in OnMiscompare() 50 LOG(INFO) << "actual: " << ShapeUtil::HumanString(actual.shape()) << " " in OnMiscompare()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_compilation_cache.cc | 71 string XlaCompilationCache::Signature::HumanString() const { in HumanString() function in tensorflow::XlaCompilationCache::Signature 134 arg.HumanString()); in BuildSignature() 240 VLOG(2) << i << ": " << args[i].HumanString(); in CompileImpl() 245 VLOG(2) << "Signature: " << signature.HumanString(); in CompileImpl() 292 << " signature: " << signature.HumanString() << " with request count " in CompileImpl() 325 VLOG(2) << "Not compiling for signature: " << signature.HumanString(); in CompileImpl()
|
D | xla_compilation_cache_test.cc | 48 << signatures[i].HumanString() << " " << signatures[j].HumanString(); in TEST()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | stream_executor_util.cc | 137 LayoutUtil::HumanString(input), in XlaConvLayoutsToStreamExecutorLayouts() 148 LayoutUtil::HumanString(filter), in XlaConvLayoutsToStreamExecutorLayouts() 159 LayoutUtil::HumanString(output), in XlaConvLayoutsToStreamExecutorLayouts()
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | dumped_computation_to_operation_list.cc | 48 absl::StrAppend(out, ShapeUtil::HumanString(operand->shape())); in DefaultAction() 53 ShapeUtil::HumanString(hlo->shape()), path_); in DefaultAction()
|