Home
last modified time | relevance | path

Searched refs:HumanStringWithLayout (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinfeed_thunk.cc42 << ShapeUtil::HumanStringWithLayout(infeed_shape); in ExecuteOnStream()
44 << ShapeUtil::HumanStringWithLayout(infeed_shape); in ExecuteOnStream()
46 << ShapeUtil::HumanStringWithLayout(infeed_shape); in ExecuteOnStream()
50 << ShapeUtil::HumanStringWithLayout(infeed_shape.tuple_shapes(0)) in ExecuteOnStream()
52 << ShapeUtil::HumanStringWithLayout(infeed_buffers.shape()); in ExecuteOnStream()
Dgpu_layout_assignment_test.cc125 "input_shape=", ShapeUtil::HumanStringWithLayout(input_shape), in TEST_F()
126 ", result_shape=", ShapeUtil::HumanStringWithLayout(result_shape))); in TEST_F()
199 "input_shape=", ShapeUtil::HumanStringWithLayout(input_shape), in TEST_F()
200 ", result_shape=", ShapeUtil::HumanStringWithLayout(result_shape))); in TEST_F()
273 "input_shape=", ShapeUtil::HumanStringWithLayout(input_shape), in TEST_F()
274 ", result_shape=", ShapeUtil::HumanStringWithLayout(result_shape))); in TEST_F()
Dgpu_hlo_support_checker.cc37 ShapeUtil::HumanStringWithLayout(instruction->shape())); in Run()
Dfft_thunk.cc113 VLOG(3) << "Input shape: " << ShapeUtil::HumanStringWithLayout(input_shape_); in ExecuteOnStream()
115 << ShapeUtil::HumanStringWithLayout(output_shape_); in ExecuteOnStream()
Dcudnn_conv_runner.cc133 VLOG(3) << "input shape: " << ShapeUtil::HumanStringWithLayout(input_shape); in RunCudnnConvImpl()
134 VLOG(3) << "filter shape: " << ShapeUtil::HumanStringWithLayout(filter_shape); in RunCudnnConvImpl()
135 VLOG(3) << "Output shape: " << ShapeUtil::HumanStringWithLayout(output_shape); in RunCudnnConvImpl()
Dgpu_hlo_support_checker_test.cc71 HasSubstr(ShapeUtil::HumanStringWithLayout(sparse_shape))); in TEST_F()
Dgpu_transfer_manager.cc133 CHECK(shape.IsTuple()) << ShapeUtil::HumanStringWithLayout(shape); in ShapeTreeToLiteral()
/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc325 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), "f32[2,3,4]{0,1,2}"); in TEST_F()
331 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
338 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
349 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
357 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
366 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
372 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
381 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
391 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
Dshape_layout.h80 string ToString() const { return ShapeUtil::HumanStringWithLayout(shape_); } in ToString()
Dshape_util.cc447 /* static */ string ShapeUtil::HumanStringWithLayout(const Shape& shape) { in HumanStringWithLayout() function in xla::ShapeUtil
452 StrAppend(&text, prefix, HumanStringWithLayout(elem_shape)); in HumanStringWithLayout()
981 << "shape=" << HumanStringWithLayout(shape) in PermuteDimensions()
982 << ", new_shape=" << HumanStringWithLayout(new_shape) in PermuteDimensions()
1404 << ShapeUtil::HumanStringWithLayout(input_shape) in AlignLayouts()
1406 << ShapeUtil::HumanStringWithLayout(output_shape_with_layout); in AlignLayouts()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.cc80 "), on-host shape=" + ShapeUtil::HumanStringWithLayout(on_host_shape()), in ToString()
82 ShapeUtil::HumanStringWithLayout(on_device_shape()), in ToString()
91 shape_str = ShapeUtil::HumanStringWithLayout(subshape); in ToString()
Dhlo_matchers.cc174 << ShapeUtil::HumanStringWithLayout(shape_) << ")"; in MatchAndExplain()
179 *os << ShapeUtil::HumanStringWithLayout(shape_); in DescribeTo()
Dhlo_module.cc292 << ShapeUtil::HumanStringWithLayout( in CreateFromProto()
294 << ", actual: " << ShapeUtil::HumanStringWithLayout(parameter_shape); in CreateFromProto()
302 << ShapeUtil::HumanStringWithLayout(expected_program_shape.result()) in CreateFromProto()
303 << ", actual: " << ShapeUtil::HumanStringWithLayout(result_shape); in CreateFromProto()
Dhlo_input_output_alias_config.cc208 ShapeUtil::HumanStringWithLayout(param_subshape), in Verify()
210 ShapeUtil::HumanStringWithLayout(output_subshape), in Verify()
Doptimize_input_output_buffer_alias.cc83 << ShapeUtil::HumanStringWithLayout(subshape) << " at index " in Build()
Dlayout_assignment.cc204 << ShapeUtil::HumanStringWithLayout(shape_with_layout); in SetOperandLayout()
220 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetOperandLayout()
265 << ShapeUtil::HumanStringWithLayout(shape_with_layout); in SetResultLayout()
274 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetResultLayout()
291 << ShapeUtil::HumanStringWithLayout(shape_with_layout); in SetInstructionLayout()
297 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetInstructionLayout()
765 ShapeUtil::HumanStringWithLayout(constant->literal().shape())); in CheckConstantLayout()
905 ShapeUtil::HumanStringWithLayout(instruction_subshape), in CheckLayouts()
906 ShapeUtil::HumanStringWithLayout(buffer->shape())); in CheckLayouts()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dinteractive_graphviz.cc269 << ShapeUtil::HumanStringWithLayout(param->shape()) << ")" in DoInfoCommand()
274 << ShapeUtil::HumanStringWithLayout(root->shape()) << ")" in DoInfoCommand()
302 std::cout << " Shape: " << ShapeUtil::HumanStringWithLayout(instr->shape()) in DoInfoCommand()
326 << ShapeUtil::HumanStringWithLayout(operand->shape()) << ")" in DoInfoCommand()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc103 << xla::ShapeUtil::HumanStringWithLayout(shape) << " device_shape=" in AllocateScopedShapedBuffer()
104 << xla::ShapeUtil::HumanStringWithLayout(on_device_shape); in AllocateScopedShapedBuffer()
246 xla::ShapeUtil::HumanStringWithLayout(literal.shape()), in WriteLiteral()
247 " device=", xla::ShapeUtil::HumanStringWithLayout(on_host_shape())); in WriteLiteral()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_hlo_support_checker.cc37 ShapeUtil::HumanStringWithLayout(instruction->shape())); in Run()
Dcpu_hlo_support_checker_test.cc71 HasSubstr(ShapeUtil::HumanStringWithLayout(sparse_shape))); in TEST_F()
Dcpu_runtime.cc100 return xla::ShapeUtil::HumanStringWithLayout(shape.ValueOrDie()); in ShapeString()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc160 << "] = " << xla::ShapeUtil::HumanStringWithLayout(shapes.back()); in GetShapeWithLayout()
179 << xla::ShapeUtil::HumanStringWithLayout(*output_shape); in GetShapeWithLayout()
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.cc70 result_layout = ShapeUtil::HumanStringWithLayout(result_layout_); in ToString()
Dlocal_client.cc75 ShapeUtil::HumanStringWithLayout( in ValidateExecutionOptions()
77 ShapeUtil::HumanStringWithLayout(arguments[i]->on_host_shape())); in ValidateExecutionOptions()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc205 ShapeUtil::HumanStringWithLayout(layout))); in ComputeAndCompareLiteralWithAllOutputLayouts()
230 ShapeUtil::HumanStringWithLayout(literal.shape())); in ComputeAndCompareLiteralWithAllInputLayouts()
244 ShapeUtil::HumanStringWithLayout(literal_relayout.shape())); in ComputeAndCompareLiteralWithAllInputLayouts()

12