| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_conv_padding_legalization.cc | 375 << ShapeUtil::HumanString(slice_shape) << " vs "
376 << ShapeUtil::HumanString(backward_conv_shape);
|
| cudnn_conv_runner.cc | 150 << ShapeUtil::HumanString(output_shape);
429 LOG(FATAL) << ShapeUtil::HumanString(*params.output_shape);
|
| gpu_executable.cc | 224 << ShapeUtil::HumanString(literal.shape());
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| layout_assignment.cc | 74 LayoutUtil::HumanString(layout_));
158 << LayoutUtil::HumanString(layout);
165 buffer.ToString(), ShapeUtil::HumanString(buffer.shape()));
181 buffer.ToString(), LayoutUtil::HumanString(curr_constraint.layout()),
182 LayoutUtil::HumanString(layout));
296 instruction->name(), ShapeUtil::HumanString(instruction->shape()),
370 LayoutUtil::HumanString(*BufferLayout(*buffer)), "\n");
411 << LayoutUtil::HumanString(*prev_layout) << ": "
[all...] |
| hlo_input_output_alias_config.cc | 35 ShapeUtil::HumanString(alias_.shape()));
|
| hlo_value.cc | 284 StrCat("InstructionValueSet(", ShapeUtil::HumanString(shape()), ")\n");
|
| hlo_verifier.cc | [all...] |
| hlo_computation.cc | 490 s << ShapeUtil::HumanString(ComputeProgramShape()) << " ";
670 instruction->name(), ShapeUtil::HumanString(instruction->shape()),
671 ShapeUtil::HumanString(indices_to_copy->shape()));
766 << ShapeUtil::HumanString(old_instruction->shape()) << " vs "
767 << ShapeUtil::HumanString(new_instruction->shape());
[all...] |
| hlo_evaluator_typed_visitor.h | [all...] |
| hlo_verifier.h | 133 : ShapeUtil::HumanString(s);
|
| reshape_mover.cc | 197 << ShapeUtil::HumanString(new_operand_shape);
|
| service.cc | 224 ShapeUtil::HumanString(result_shape));
284 i, ShapeUtil::HumanString(program_shape.parameters(i)),
285 ShapeUtil::HumanString(*argument_shapes[i]));
[all...] |
| indexed_array_analysis.cc | 50 return absl::StrCat("(constant ", ShapeUtil::HumanString(root->shape()),
53 return absl::StrCat("(constant ", ShapeUtil::HumanString(root->shape()),
61 " to ", ShapeUtil::HumanString(reshaped_array->shape()), ")");
[all...] |
| while_loop_simplifier.cc | 585 CHECK(desired_shape.IsTuple()) << ShapeUtil::HumanString(desired_shape);
691 << ShapeUtil::HumanString(nested_instr->shape()) << " vs "
692 << ShapeUtil::HumanString(while_shape);
[all...] |
| hlo_evaluator.cc | 251 ShapeUtil::HumanString(arg_shape));
426 << "parameter shape is: " << ShapeUtil::HumanString(parameter->shape())
428 << ShapeUtil::HumanString(input_literal->shape());
[all...] |
| pattern_matcher.h | 412 EXPLAIN << "Layout " << LayoutUtil::HumanString(*layout)
414 << LayoutUtil::HumanString(*layout_);
421 *os << "equal to " << LayoutUtil::HumanString(*layout_);
717 << ShapeUtil::HumanString(*shape_);
724 *os << "compatible with " << ShapeUtil::HumanString(*shape_);
[all...] |
| hlo_graph_dumper.cc | 824 return StrFormat("{} (%s)", ShapeUtil::HumanString(constant->shape()));
850 ShapeUtil::HumanString(constant->shape()));
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batchtospace_op.cc | 47 xla::ShapeUtil::HumanString(crops.shape())));
|
| spacetobatch_op.cc | 47 xla::ShapeUtil::HumanString(paddings.shape())));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_compiler.h | 169 string HumanString() const;
|
| xla_op_kernel.cc | 56 << expression->HumanString();
66 << expression->HumanString();
301 xla::ShapeUtil::HumanString(literal.shape()));
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util.h | 236 static string HumanString(const Shape& shape);
242 static string HumanString(const ProgramShape& program_shape);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| test_utils.cc | 267 ShapeUtil::HumanString(shape));
340 ShapeUtil::HumanString(shape));
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| local_client.cc | 332 ::xla::ShapeUtil::HumanString(shape))));
|
| client.cc | 319 << ShapeUtil::HumanString(shape);
|
