/external/tensorflow/tensorflow/compiler/xla/service/ |
D | shape_inference.cc | 174 ShapeUtil::HumanString(shape).c_str()); in ExpectNotTupleOrOpaque() 178 ShapeUtil::HumanString(shape).c_str()); in ExpectNotTupleOrOpaque() 207 ShapeUtil::HumanString(reducer_shape.parameters(0)).c_str(), in VerifyReducerShape() 208 ShapeUtil::HumanString(accumulator_shape).c_str()); in VerifyReducerShape() 217 ShapeUtil::HumanString(accumulator_shape).c_str(), in VerifyReducerShape() 218 ShapeUtil::HumanString(init_value_shape).c_str()); in VerifyReducerShape() 229 ShapeUtil::HumanString(reducer_shape.parameters(1)).c_str(), in VerifyReducerShape() 230 ShapeUtil::HumanString(input_element_shape).c_str()); in VerifyReducerShape() 240 ShapeUtil::HumanString(reducer_shape.parameters(1)).c_str(), in VerifyReducerShape() 241 ShapeUtil::HumanString(accumulator_shape).c_str()); in VerifyReducerShape() [all …]
|
D | hlo_evaluator.cc | 156 ShapeUtil::HumanString(shape).c_str(), in ElementWiseUnaryOpImpl() 157 ShapeUtil::HumanString(operand->shape()).c_str()); in ElementWiseUnaryOpImpl() 867 << "return shape set to: " << ShapeUtil::HumanString(result_shape) in HandleReverse() 869 << ShapeUtil::HumanString(inferred_return_shape); in HandleReverse() 919 << "return shape set to: " << ShapeUtil::HumanString(result_shape) in HandleConvolution() 921 << ShapeUtil::HumanString(inferred_return_shape); in HandleConvolution() 1172 << "return shape is set to: " << ShapeUtil::HumanString(pad->shape()) in HandlePad() 1174 << ShapeUtil::HumanString(inferred_return_shape); in HandlePad() 1238 << "return shape is set to: " << ShapeUtil::HumanString(result_shape) in HandleDynamicSlice() 1240 << ShapeUtil::HumanString(inferred_return_shape); in HandleDynamicSlice() [all …]
|
D | transfer_manager.cc | 49 << ShapeUtil::HumanString(literal.shape()) in TransferArrayToDevice() 50 << " is not an array: " << ShapeUtil::HumanString(on_device_shape); in TransferArrayToDevice() 67 << "Shape " << ShapeUtil::HumanString(shape) in TransferArrayFromDevice() 69 << ShapeUtil::HumanString(HostShapeToDeviceShape(shape)); in TransferArrayFromDevice()
|
D | layout_assignment.cc | 67 << ShapeUtil::HumanString(shape_with_layout) << " " in CreateCopyWithNewLayout() 68 << ShapeUtil::HumanString(instruction->shape()) in CreateCopyWithNewLayout() 149 LayoutUtil::HumanString(layout_).c_str()); in ToString() 216 << LayoutUtil::HumanString(layout); in SetBufferLayout() 224 ShapeUtil::HumanString(buffer.shape()).c_str()); in SetBufferLayout() 241 LayoutUtil::HumanString(curr_constraint->layout()).c_str(), in SetBufferLayout() 242 LayoutUtil::HumanString(layout).c_str()); in SetBufferLayout() 366 ShapeUtil::HumanString(instruction->shape()).c_str(), in SetInstructionLayout() 443 LayoutUtil::HumanString(*BufferLayout(*buffer)), "\n"); in ToString() 572 ShapeUtil::HumanString(body_layout.result_shape()).c_str(), in AddMandatoryConstraints() [all …]
|
D | local_service.cc | 112 ShapeUtil::HumanString(program_shape->parameters(i)).c_str(), in CompileExecutable() 113 ShapeUtil::HumanString(argument_shape).c_str()); in CompileExecutable()
|
D | hlo_sharding.cc | 60 return StrCat("{", ShapeUtil::HumanString(tile_shape_), " ", in ToString() 205 " against shape ", ShapeUtil::HumanString(shape))); in Validate() 250 ", input_shape=", ShapeUtil::HumanString(shape)); in ValidateNonTuple()
|
/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
D | batch_dot.cc | 42 xla::ShapeUtil::HumanString(*x_shape), " vs. ", in BatchDot() 43 xla::ShapeUtil::HumanString(*y_shape)); in BatchDot() 58 xla::ShapeUtil::HumanString(*x_shape), " vs ", in BatchDot() 59 xla::ShapeUtil::HumanString(*y_shape)); in BatchDot() 70 xla::ShapeUtil::HumanString(*x_shape), " transpose: ", transpose_x, in BatchDot() 71 " vs. ", xla::ShapeUtil::HumanString(*y_shape), in BatchDot()
|
D | scatter.cc | 61 xla::ShapeUtil::HumanString(*indices_shape), in XlaScatter() 63 xla::ShapeUtil::HumanString(*buffer_shape), ")"); in XlaScatter() 84 xla::ShapeUtil::HumanString(*buffer_shape)); in XlaScatter()
|
D | triangular_solve.cc | 43 xla::ShapeUtil::HumanString(*a_shape), " vs. ", in TriangularSolve() 44 xla::ShapeUtil::HumanString(*b_shape)); in TriangularSolve() 59 xla::ShapeUtil::HumanString(*a_shape), " vs ", in TriangularSolve() 60 xla::ShapeUtil::HumanString(*b_shape)); in TriangularSolve() 69 xla::ShapeUtil::HumanString(*a_shape)); in TriangularSolve() 76 xla::ShapeUtil::HumanString(*a_shape), " vs ", in TriangularSolve() 77 xla::ShapeUtil::HumanString(*b_shape)); in TriangularSolve()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | shape_util_test.cc | 79 << "expected: " << ShapeUtil::HumanString(expected) in TEST() 80 << "actual: " << ShapeUtil::HumanString(actual); in TEST() 91 << "expected: " << ShapeUtil::HumanString(expected) in TEST() 92 << "actual: " << ShapeUtil::HumanString(actual); in TEST() 105 << "expected: " << ShapeUtil::HumanString(expected) in TEST() 106 << "actual: " << ShapeUtil::HumanString(actual); in TEST() 115 << "expected: " << ShapeUtil::HumanString(expected) in TEST() 116 << "actual: " << ShapeUtil::HumanString(actual); in TEST() 125 << "expected: " << ShapeUtil::HumanString(expected) in TEST() 126 << "actual: " << ShapeUtil::HumanString(actual); in TEST() [all …]
|
D | shape_layout.cc | 28 ShapeUtil::HumanString(other_shape).c_str(), in CopyLayoutFromShape() 29 ShapeUtil::HumanString(shape()).c_str()); in CopyLayoutFromShape() 38 ShapeUtil::HumanString(*to_shape).c_str(), in AssignLayoutToShape() 39 ShapeUtil::HumanString(shape()).c_str()); in AssignLayoutToShape()
|
D | packed_literal_reader.cc | 44 VLOG(3) << "reading shape from file: " << ShapeUtil::HumanString(shape) in Read() 80 VLOG(3) << "read shape from file: " << ShapeUtil::HumanString(shape); in Read()
|
D | layout_util.cc | 162 ShapeUtil::HumanString(shape).c_str()); in ValidateLayoutInShape() 200 HumanString(layout).c_str()); in ValidateLayoutForShape() 205 HumanString(layout).c_str()); in ValidateLayoutForShape() 373 /* static */ string LayoutUtil::HumanString(const Layout& layout) { in HumanString() function in xla::LayoutUtil 464 out << LayoutUtil::HumanString(layout); in operator <<()
|
D | shape_util.cc | 334 CHECK(IsTuple(shape)) << HumanString(shape); in TupleElementCount() 361 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape() 373 CHECK(!IsTuple(shape)) << ShapeUtil::HumanString(shape); in ElementsIn() 388 /* static */ string ShapeUtil::HumanString(const Shape& shape) { in HumanString() function in xla::ShapeUtil 393 tensorflow::strings::StrAppend(&text, prefix, HumanString(elem_shape)); in HumanString() 477 LayoutUtil::HumanString(shape.layout())); in HumanStringWithLayout() 484 /* static */ string ShapeUtil::HumanString(const ProgramShape& program_shape) { in HumanString() function in xla::ShapeUtil 492 ": ", HumanString(shape))); in HumanString() 496 HumanString(program_shape.result())); in HumanString() 1439 out << ShapeUtil::HumanString(shape); in operator <<()
|
D | literal_util.cc | 363 ShapeUtil::HumanString(dest_subshape).c_str(), in CopyFrom() 364 ShapeUtil::HumanString(src_subshape).c_str()); in CopyFrom() 405 ShapeUtil::HumanString(dest_subshape).c_str(), in MoveFrom() 406 ShapeUtil::HumanString(src_literal.shape()).c_str()); in MoveFrom() 443 TF_RET_CHECK(ShapeUtil::IsArray(shape())) << ShapeUtil::HumanString(shape()); in CopySliceFrom() 445 << ShapeUtil::HumanString(src_literal.shape()); in CopySliceFrom() 720 << "Given shape_with_layout " << ShapeUtil::HumanString(shape_with_layout) in Relayout() 722 << ShapeUtil::HumanString(shape()); in Relayout() 758 ShapeUtil::HumanString(shape()).c_str(), in Reshape() 759 ShapeUtil::HumanString(output->shape()).c_str()); in Reshape() [all …]
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | while_op.cc | 209 VLOG(2) << "Body shape: " << xla::ShapeUtil::HumanString(body_input_shape) in Compile() 210 << " -> " << xla::ShapeUtil::HumanString(body.xla_output_shape); in Compile() 211 VLOG(2) << "Cond shape: " << xla::ShapeUtil::HumanString(cond_input_shape) in Compile() 212 << " -> " << xla::ShapeUtil::HumanString(cond.xla_output_shape); in Compile() 218 xla::ShapeUtil::HumanString(body_input_shape), " vs. ", in Compile() 219 xla::ShapeUtil::HumanString(cond_input_shape))); in Compile() 224 xla::ShapeUtil::HumanString(body_input_shape), " vs. ", in Compile() 225 xla::ShapeUtil::HumanString(body.xla_output_shape))); in Compile() 234 xla::ShapeUtil::HumanString(cond.xla_output_shape))); in Compile()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_convolution_runner.cc | 85 VLOG(3) << "input shape: { " << ShapeUtil::HumanString(input_shape) << " }"; in RunCudnnConvolution() 86 VLOG(3) << "filter shape: { " << ShapeUtil::HumanString(filter_shape) << " }"; in RunCudnnConvolution() 87 VLOG(3) << "Output shape: { " << ShapeUtil::HumanString(output_shape) << " }"; in RunCudnnConvolution() 101 << ShapeUtil::HumanString(output_shape); in RunCudnnConvolution() 104 << ShapeUtil::HumanString(output_shape); in RunCudnnConvolution() 106 LOG(FATAL) << ShapeUtil::HumanString(output_shape); in RunCudnnConvolution() 245 << ShapeUtil::HumanString(output_shape); in RunCudnnConvolution()
|
D | gpu_transfer_manager.cc | 53 << ShapeUtil::HumanString(shape); in TransferLiteralToInfeed() 63 ShapeUtil::HumanString(literal.shape()).c_str()); in TransferLiteralToInfeed()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_transfer_manager.cc | 97 << ShapeUtil::HumanString(shape); in TransferLiteralToInfeed() 107 ShapeUtil::HumanString(literal.shape()).c_str()); in TransferLiteralToInfeed() 193 << ShapeUtil::HumanString(received_shape) in TransferLiteralFromOutfeed() 195 << ShapeUtil::HumanString(literal_shape); in TransferLiteralFromOutfeed() 229 << ShapeUtil::HumanString(received_shape) in TransferLiteralFromOutfeed() 231 << ShapeUtil::HumanString(literal_shape); in TransferLiteralFromOutfeed()
|
D | xfeed_manager_test.cc | 49 << "want " << ShapeUtil::HumanString(shape_) << " " in Done() 51 << ShapeUtil::HumanString(shape.ValueOrDie()); in Done()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | while_test.cc | 233 VLOG(2) << "while = " << ShapeUtil::HumanString( in TEST_F() 286 VLOG(2) << "while = " << ShapeUtil::HumanString( in TEST_F() 346 << ShapeUtil::HumanString( in TEST_F() 356 VLOG(2) << "expected = " << ShapeUtil::HumanString(expected->shape()); in TEST_F() 400 << ShapeUtil::HumanString( in TEST_F() 409 VLOG(2) << "expected = " << ShapeUtil::HumanString(expected->shape()); in TEST_F() 457 << ShapeUtil::HumanString( in TEST_F() 507 VLOG(2) << "while = " << ShapeUtil::HumanString( in TEST_F() 515 VLOG(2) << "expected = " << ShapeUtil::HumanString(expected->shape()); in TEST_F() 556 << ShapeUtil::HumanString( in TEST_F() [all …]
|
D | literal_test_util.cc | 46 << "tupleness-mismatch! want: " << ShapeUtil::HumanString(expected) in EqualShapes() 47 << " got: " << ShapeUtil::HumanString(actual); in EqualShapes() 69 << "want rank of: " << ShapeUtil::HumanString(expected) in EqualShapes() 70 << " got rank of: " << ShapeUtil::HumanString(actual); in EqualShapes() 86 << " expected: " << ShapeUtil::HumanString(expected) in EqualShapes() 87 << " actual: " << ShapeUtil::HumanString(actual); in EqualShapes() 321 ShapeUtil::HumanString(expected.shape()))); in Equal() 411 LOG(INFO) << "expected: " << ShapeUtil::HumanString(expected.shape()) in ExpectNear() 413 LOG(INFO) << "actual: " << ShapeUtil::HumanString(actual.shape()) in ExpectNear() 699 "Tuple index ", i, " in ", ShapeUtil::HumanString(expected.shape()))); in Near()
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | dumped_computation_to_operation_list.cc | 50 out, ShapeUtil::HumanString(operand->shape())); in DefaultAction() 55 params.c_str(), ShapeUtil::HumanString(hlo->shape()).c_str(), in DefaultAction()
|
D | replay_computation.cc | 163 ShapeUtil::HumanString(result->shape()).c_str(), in RealMain() 169 ShapeUtil::HumanString(module.result().shape()).c_str(), in RealMain()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | computation_builder.cc | 260 << "want " << ShapeUtil::HumanString(expected_shape) << " got " in CheckShape() 261 << ShapeUtil::HumanString(*actual_shape); in CheckShape() 269 VLOG(2) << "checking " << ShapeUtil::HumanString(*lhs_shape) << " equals " in CheckSameShape() 270 << ShapeUtil::HumanString(*rhs_shape); in CheckSameShape() 272 << "lhs " << ShapeUtil::HumanString(*lhs_shape) << " rhs " in CheckSameShape() 273 << ShapeUtil::HumanString(*rhs_shape); in CheckSameShape() 435 VLOG(3) << "original shape: " << ShapeUtil::HumanString(*original_shape); in Collapse() 581 ShapeUtil::HumanString(lhs_shape).c_str(), in VerifyConvolution() 582 ShapeUtil::HumanString(rhs_shape).c_str())); in VerifyConvolution() 590 ShapeUtil::HumanString(lhs_shape).c_str(), in VerifyConvolution() [all …]
|