| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_hlo_support_checker.cc | 38 ShapeUtil::HumanStringWithLayout(instruction->shape())
|
| cpu_hlo_support_checker_test.cc | 68 HasSubstr(ShapeUtil::HumanStringWithLayout(sparse_shape)));
|
| cpu_runtime.cc | 73 return xla::ShapeUtil::HumanStringWithLayout(shape.ValueOrDie());
|
| ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_hlo_support_checker.cc | 38 ShapeUtil::HumanStringWithLayout(instruction->shape())
|
| gpu_layout_assignment_test.cc | 118 "input_shape=", ShapeUtil::HumanStringWithLayout(input_shape),
119 ", result_shape=", ShapeUtil::HumanStringWithLayout(result_shape)));
190 "input_shape=", ShapeUtil::HumanStringWithLayout(input_shape),
191 ", result_shape=", ShapeUtil::HumanStringWithLayout(result_shape)));
262 "input_shape=", ShapeUtil::HumanStringWithLayout(input_shape),
263 ", result_shape=", ShapeUtil::HumanStringWithLayout(result_shape)));
|
| gpu_hlo_support_checker_test.cc | 68 HasSubstr(ShapeUtil::HumanStringWithLayout(sparse_shape)));
|
| fft_thunk.cc | 129 VLOG(3) << "Input shape: " << ShapeUtil::HumanStringWithLayout(input_shape_);
131 << ShapeUtil::HumanStringWithLayout(output_shape_);
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_layout.h | 77 string ToString() const { return ShapeUtil::HumanStringWithLayout(shape_); }
|
| shape_util_test.cc | 264 << ShapeUtil::HumanStringWithLayout(scalar_default_layout);
269 << ShapeUtil::HumanStringWithLayout(scalar_empty_min2maj);
455 EXPECT_EQ("opaque[]", ShapeUtil::HumanStringWithLayout(opaque));
456 EXPECT_EQ("f32[]", ShapeUtil::HumanStringWithLayout(scalar));
457 EXPECT_EQ("u32[1,2]{1,0}", ShapeUtil::HumanStringWithLayout(matrix));
458 EXPECT_EQ("s32[3,4]{0,1}", ShapeUtil::HumanStringWithLayout(matrix2));
460 ShapeUtil::HumanStringWithLayout(tuple));
462 ShapeUtil::HumanStringWithLayout(nested_tuple));
|
| shape_util.cc | 455 /* static */ string ShapeUtil::HumanStringWithLayout(const Shape& shape) {
461 HumanStringWithLayout(elem_shape));
890 VLOG(10) << "Original_shape: " << HumanStringWithLayout(shape);
891 VLOG(10) << "Stripped_shape: " << HumanStringWithLayout(stripped_shape);
[all...] |
| shape_util.h | 198 static string HumanStringWithLayout(const Shape& shape);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| executable_build_options.cc | 56 result_layout = ShapeUtil::HumanStringWithLayout(result_layout_);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| shaped_buffer.cc | 81 "), on-host shape=" + ShapeUtil::HumanStringWithLayout(on_host_shape()),
83 ShapeUtil::HumanStringWithLayout(on_device_shape()),
92 shape_str = ShapeUtil::HumanStringWithLayout(subshape);
|
| hlo_module.cc | 293 << ShapeUtil::HumanStringWithLayout(
295 << ", actual: " << ShapeUtil::HumanStringWithLayout(parameter_shape);
302 << ShapeUtil::HumanStringWithLayout(expected_program_shape.result())
303 << ", actual: " << ShapeUtil::HumanStringWithLayout(result_shape);
|
| layout_assignment.cc | 273 << ShapeUtil::HumanStringWithLayout(shape_with_layout);
289 ShapeUtil::HumanStringWithLayout(shape_with_layout).c_str());
334 << ShapeUtil::HumanStringWithLayout(shape_with_layout);
343 ShapeUtil::HumanStringWithLayout(shape_with_layout).c_str());
360 << ShapeUtil::HumanStringWithLayout(shape_with_layout);
367 ShapeUtil::HumanStringWithLayout(shape_with_layout).c_str());
[all...] |
| hlo_tfgraph_builder.cc | 166 layout_string = ShapeUtil::HumanStringWithLayout(instruction->shape());
|
| transfer_manager.cc | 189 ShapeUtil::HumanStringWithLayout(on_host_shape).c_str());
|
| service.cc | 224 ShapeUtil::HumanStringWithLayout(shape_with_layout).c_str(),
230 ShapeUtil::HumanStringWithLayout(shape_with_layout).c_str());
[all...] |
| hlo_graph_dumper.cc | [all...] |
| buffer_assignment.cc | 153 ShapeUtil::HumanStringWithLayout(buffer->shape()).c_str()));
[all...] |
| hlo_evaluator.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| client_library_test_base.cc | 182 ShapeUtil::HumanStringWithLayout(layout)));
208 ShapeUtil::HumanStringWithLayout(literal->shape()));
222 ShapeUtil::HumanStringWithLayout(literal_relayout->shape()));
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| xla_launch_op.cc | 298 << xla::ShapeUtil::HumanStringWithLayout(on_device_shape)
300 << xla::ShapeUtil::HumanStringWithLayout(shape);
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| ir_array.cc | 39 << "Shape " << ShapeUtil::HumanStringWithLayout(shape)
75 << "Shape " << ShapeUtil::HumanStringWithLayout(shape)
|
