/external/tensorflow/tensorflow/compiler/xla/ |
shape_layout.cc | 28 ShapeUtil::HumanString(other_shape), 29 ShapeUtil::HumanString(shape())); 38 ShapeUtil::HumanString(*to_shape), 39 ShapeUtil::HumanString(shape()));
|
packed_literal_reader.cc | 44 VLOG(3) << "reading shape from file: " << ShapeUtil::HumanString(shape) 79 VLOG(3) << "read shape from file: " << ShapeUtil::HumanString(shape);
|
text_literal_writer.cc | 40 s = f->Append(ShapeUtil::HumanString(literal.shape()) + "\n");
|
/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
scatter.cc | 54 xla::ShapeUtil::HumanString(indices_shape), 56 xla::ShapeUtil::HumanString(buffer_shape), ")"); 77 xla::ShapeUtil::HumanString(buffer_shape)); 189 VLOG(3) << " Input: " << xla::ShapeUtil::HumanString(buffer_shape); 190 VLOG(3) << " Indices: " << xla::ShapeUtil::HumanString(indices_shape); 191 VLOG(3) << " Updates: " << xla::ShapeUtil::HumanString(updates_shape);
|
broadcast.cc | 41 "Input shape (", xla::ShapeUtil::HumanString(input_shape), 55 xla::ShapeUtil::HumanString(input_shape),
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
xla_expression_test.cc | 70 TEST_F(XlaExpressionTest, HumanString) { 71 EXPECT_EQ("invalid", XlaExpression().HumanString()); 72 EXPECT_EQ("invalid", XlaExpression::Invalid().HumanString()); 73 EXPECT_EQ("constant", XlaExpression::Constant(constant_).HumanString()); 74 EXPECT_EQ("xla_op", XlaExpression::XlaOp(op_, DT_INT32).HumanString()); 75 EXPECT_EQ("resource", XlaExpression::Resource(resource_.get()).HumanString());
|
xla_expression.cc | 65 string XlaExpression::HumanString() const { 99 HumanString()); 117 "ResolveConstant called on XlaExpression: ", HumanString());
|
xla_expression.h | 89 string HumanString() const;
|
/external/tensorflow/tensorflow/compiler/jit/ |
xla_compilation_cache_test.cc | 48 << signatures[i].HumanString() << " " << signatures[j].HumanString();
|
xla_compilation_cache.cc | 71 string XlaCompilationCache::Signature::HumanString() const { 134 arg.HumanString()); 240 VLOG(2) << i << ": " << args[i].HumanString(); 245 VLOG(2) << "Signature: " << signature.HumanString(); 292 << " signature: " << signature.HumanString() << " with request count " 325 VLOG(2) << "Not compiling for signature: " << signature.HumanString();
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
shape_inference.cc | 55 string(op_type), ShapeUtil::HumanString(shape)); 95 ShapeUtil::HumanString(accumulator_shape)); 103 ShapeUtil::HumanString(accumulator_shape)); 116 i, ShapeUtil::HumanString(reducer_shape.parameters(i)), 117 ShapeUtil::HumanString(*accumulator_subshapes[i])); 125 i, ShapeUtil::HumanString(*accumulator_subshapes[i]), 126 ShapeUtil::HumanString(*init_value_shapes[i])); 137 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)), 138 ShapeUtil::HumanString(input_element_shape)); 151 ShapeUtil::HumanString(reducer_shape.parameters(inputs + i)) [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
if_op.cc | 83 << " shape: " << arg.HumanString() 92 << " shape: " << arg.HumanString(); 164 xla::ShapeUtil::HumanString(then_input_shape), " vs. ", 165 xla::ShapeUtil::HumanString(else_input_shape))); 174 xla::ShapeUtil::HumanString(then_result.xla_output_shape), " vs. ", 175 xla::ShapeUtil::HumanString(else_result.xla_output_shape))); 177 VLOG(2) << "Input shape: " << xla::ShapeUtil::HumanString(then_input_shape); 179 << xla::ShapeUtil::HumanString(then_result.xla_output_shape); 234 << xla::ShapeUtil::HumanString(shape_or.ValueOrDie()); 254 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))) [all...] |
xla_select_and_scatter_op.cc | 85 xla::ShapeUtil::HumanString(select_output_shape), " got ", 86 xla::ShapeUtil::HumanString(select.xla_output_shape))); 107 xla::ShapeUtil::HumanString(scalar_shape), " got ", 108 xla::ShapeUtil::HumanString(scatter.xla_output_shape)));
|
case_op.cc | 80 << " shape: " << arg.HumanString() 89 << " shape: " << arg.HumanString(); 168 xla::ShapeUtil::HumanString(branch0_input_shape), " vs. ", 169 xla::ShapeUtil::HumanString(branch_input_shape))); 178 xla::ShapeUtil::HumanString(branch_results[0].xla_output_shape), 180 xla::ShapeUtil::HumanString(branch_results[j].xla_output_shape))); 184 << xla::ShapeUtil::HumanString(branch0_input_shape); 186 << xla::ShapeUtil::HumanString( 246 << xla::ShapeUtil::HumanString(shape_or.ValueOrDie()); 266 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))) [all...] |
xla_reduce_op.cc | 83 xla::ShapeUtil::HumanString(scalar_shape), " got ", 84 xla::ShapeUtil::HumanString(reducer.xla_output_shape)));
|
while_op.cc | 173 xla::ShapeUtil::HumanString(body_input_shape), " vs. ", 174 xla::ShapeUtil::HumanString(body_output_shape)); 428 VLOG(2) << "Body shape: " << xla::ShapeUtil::HumanString(body_input_shape) 429 << " -> " << xla::ShapeUtil::HumanString(body.xla_output_shape); 430 VLOG(2) << "Cond shape: " << xla::ShapeUtil::HumanString(cond_input_shape) 431 << " -> " << xla::ShapeUtil::HumanString(cond.xla_output_shape); 437 xla::ShapeUtil::HumanString(body_input_shape), " vs. ", 438 xla::ShapeUtil::HumanString(cond_input_shape))); 463 xla::ShapeUtil::HumanString(cond.xla_output_shape))); 548 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))) [all...] |
/external/tensorflow/tensorflow/compiler/xla/tools/ |
show_signature.cc | 58 ShapeUtil::HumanString(*shape).c_str());
|
dumped_computation_to_operation_list.cc | 48 absl::StrAppend(out, ShapeUtil::HumanString(operand->shape())); 53 ShapeUtil::HumanString(hlo->shape()), path_);
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
stream_executor_util.cc | 137 LayoutUtil::HumanString(input), 148 LayoutUtil::HumanString(filter), 159 LayoutUtil::HumanString(output),
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
cpu_transfer_manager.cc | 97 << ShapeUtil::HumanString(shape); 107 ShapeUtil::HumanString(literal.shape())); 192 << ShapeUtil::HumanString(received_shape) 194 << ShapeUtil::HumanString(literal_shape); 218 << ShapeUtil::HumanString(received_shape) 220 << ShapeUtil::HumanString(literal_shape);
|
xfeed_manager.cc | 73 << (shape.ok() ? ShapeUtil::HumanString(shape.ValueOrDie())
|
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
executable.cc | 78 ShapeUtil::HumanString(expected_shape), 79 ShapeUtil::HumanString(actual_shape));
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
literal_test_util.cc | 48 LOG(INFO) << "expected: " << ShapeUtil::HumanString(expected.shape()) << " " 50 LOG(INFO) << "actual: " << ShapeUtil::HumanString(actual.shape()) << " "
|
axpy_simple_test.cc | 67 EXPECT_EQ("() -> f32[10]", ShapeUtil::HumanString(shape));
|
test_utils_test.cc | 251 << ShapeUtil::HumanString(args[0].shape()); 254 << ShapeUtil::HumanString(args[1].shape()); 283 << ShapeUtil::HumanString(indices_shape); 321 << ShapeUtil::HumanString(indices_shape);
|