| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| local_client_aot_test_helper.cc | 39 auto r0f32 = xla::ShapeUtil::MakeShape(xla::F32, {});
53 auto opaque_shape = xla::ShapeUtil::MakeOpaqueShape();
55 auto r0f32 = xla::ShapeUtil::MakeShape(xla::F32, {});
|
| prng_test.cc | 56 ShapeUtil::MakeShape(primitive_util::NativeToPrimitiveType<T>(), dims));
147 ShapeUtil::MakeShape(S32, {sample_size}));
188 auto x = Parameter(b.get(), 0, ShapeUtil::MakeShape(F32, {}), "input");
191 ShapeUtil::MakeShape(F32, {})));
214 EXPECT_EQ(ShapeUtil::ElementsIn(actual.shape()),
215 ShapeUtil::ElementsIn(param0_literal.shape()));
216 for (int i = 0; i < ShapeUtil::ElementsIn(actual.shape()); ++i) {
231 ShapeUtil::MakeShape(F32, {10}));
291 ShapeUtil::MakeShape(S32, {10}));
294 ShapeUtil::MakeShape(S32, {10}))
[all...] |
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_debug.cc | 87 if (xla::ShapeUtil::TupleElementCount(padded_shape) != 2) {
99 xla::Shape shape0 = xla::ShapeUtil::GetTupleElementShape(padded_shape, 0);
101 xla::ShapeUtil::GetTupleElementShape(padded_shape, 1);
108 if (!xla::ShapeUtil::Equal(shape0, shape1)) {
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| executable_build_options.cc | 70 result_layout = ShapeUtil::HumanStringWithLayout(result_layout_);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_external_constants_test.cc | 39 auto shape = ShapeUtil::MakeShape(F32, {rows, cols});
|
| cpu_eigen_dot_operation_test.cc | 65 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128});
80 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128});
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| text_literal_writer.cc | 40 s = f->Append(ShapeUtil::HumanString(literal.shape()) + "\n");
|
| literal_comparison.cc | 190 const int64 tuple_elements = ShapeUtil::TupleElementCount(shape);
193 total += RecursiveElementCount(ShapeUtil::GetTupleElementShape(shape, i));
197 return ShapeUtil::ElementsIn(shape);
322 ShapeUtil::HumanString(expected_.shape()));
325 mismatches_ = Literal(ShapeUtil::ChangeElementType(actual_.shape(), PRED));
544 int64 element_count = ShapeUtil::ElementsIn(actual_.shape());
556 ShapeUtil::HumanString(actual_.shape()),
557 ShapeUtil::ElementsIn(actual_.shape()), error_.abs, error_.rel);
718 for (int i = 0; i < ShapeUtil::TupleElementCount(expected.shape()); ++i) {
746 for (int64 i = 0; i < ShapeUtil::TupleElementCount(expected.shape()); ++i)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| show_signature.cc | 58 ShapeUtil::HumanString(*shape).c_str());
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| transfer_manager.cc | 161 << ShapeUtil::HumanString(literal.shape())
162 << " is not an array: " << ShapeUtil::HumanString(on_device_shape);
181 if (!ShapeUtil::Equal(HostShapeToDeviceShape(shape), shape)) {
182 auto error = StrCat("Shape ", ShapeUtil::HumanString(shape),
184 ShapeUtil::HumanString(HostShapeToDeviceShape(shape)));
243 return ShapeUtil::ForEachSubshapeWithStatus(
253 for (int64 i = 0; i < ShapeUtil::TupleElementCount(device_subshape);
276 i < ShapeUtil::TupleElementCount(device_buffer.on_device_shape()); ++i) {
314 ShapeUtil::HumanStringWithLayout(on_host_shape));
316 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(on_host_shape))
[all...] |
| hlo_constant_folding.cc | 127 ShapeUtil::ElementsIn(operand->shape());
131 ShapeUtil::ElementsIn(instruction->shape());
|
| hlo_query.cc | 27 ShapeUtil::IsScalarWithElementType(instruction->shape(), F32)) {
104 return instruction->IsConstant() && ShapeUtil::IsScalar(instruction->shape());
|
| pattern_matcher_gmock_test.cc | 48 Shape s = ShapeUtil::MakeShape(F32, {10, 100});
66 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {42}), "p");
|
| zero_sized_hlo_elimination_test.cc | 45 0, ShapeUtil::MakeShape(F32, {3, 0}), "zero sized param"))) {}
86 Shape op_shape = ShapeUtil::MakeShape(F32, {4, 0});
|
| copy_insertion_test.cc | 83 const Shape scalar_shape_ = ShapeUtil::MakeShape(F32, {});
91 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {}), "x"));
175 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {}), "x"));
177 HloInstruction::CreateParameter(1, ShapeUtil::MakeShape(F32, {}), "y"));
180 ShapeUtil::MakeShape(F32, {}), HloOpcode::kAdd, constant1, y));
237 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {4}), "x"));
239 ShapeUtil::MakeShape(F32, {2, 2}), HloOpcode::kBitcast, x));
262 ShapeUtil::MakeShape(F32, {2, 2}), HloOpcode::kBitcast, constant));
281 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {4}), "x"));
283 ShapeUtil::MakeShape(F32, {2, 2}), HloOpcode::kBitcast, x))
[all...] |
| allocation_tracker.cc | 79 ShapeUtil::ForEachSubshape(
112 ShapeUtil::ForEachSubshape(
153 if (ShapeUtil::IsNestedTuple(shaped_buffer->on_device_shape())) {
159 i < ShapeUtil::TupleElementCount(shaped_buffer->on_device_shape());
162 ShapeUtil::GetTupleElementShape(shaped_buffer->on_host_shape(), i),
163 ShapeUtil::GetTupleElementShape(shaped_buffer->on_device_shape(), i),
|
| hlo_matchers.cc | 155 if (ShapeUtil::Compatible(instruction->shape(), shape_)) {
159 << ShapeUtil::HumanString(shape_) << ")";
164 *os << ShapeUtil::HumanString(shape_);
170 if (ShapeUtil::Equal(instruction->shape(), shape_)) {
174 << ShapeUtil::HumanStringWithLayout(shape_) << ")";
179 *os << ShapeUtil::HumanStringWithLayout(shape_);
|
| while_loop_simplifier.cc | 89 const int64 tuple_size = ShapeUtil::TupleElementCount(while_init->shape());
177 ShapeUtil::MakeTupleShape(new_while_tuple_elem_shapes);
325 ShapeUtil::Compatible(while_init->shape(), while_body_root->shape()));
351 Shape new_while_shape = ShapeUtil::MakeTupleShape(new_while_shape_elems);
364 CHECK(ShapeUtil::Compatible(instr->shape(), while_shape));
378 CHECK(ShapeUtil::Compatible(instr->shape(), new_while_shape));
405 if (ShapeUtil::IsEmptyTuple(new_while_shape)) {
585 CHECK(desired_shape.IsTuple()) << ShapeUtil::HumanString(desired_shape);
602 ShapeUtil::ForEachSubshape(
657 ShapeUtil::Compatible(while_init->shape(), while_body_root->shape()))
[all...] |
| layout_assignment.cc | 85 CHECK(ShapeUtil::Compatible(shape_layout.shape(),
165 buffer.ToString(), ShapeUtil::HumanString(buffer.shape()));
204 << ShapeUtil::HumanStringWithLayout(shape_with_layout);
220 ShapeUtil::HumanStringWithLayout(shape_with_layout));
265 << ShapeUtil::HumanStringWithLayout(shape_with_layout);
274 ShapeUtil::HumanStringWithLayout(shape_with_layout));
291 << ShapeUtil::HumanStringWithLayout(shape_with_layout);
293 if (!ShapeUtil::Compatible(shape_with_layout, instruction->shape())) {
296 instruction->name(), ShapeUtil::HumanString(instruction->shape()),
297 ShapeUtil::HumanStringWithLayout(shape_with_layout))
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| broadcast.cc | 41 "Input shape (", xla::ShapeUtil::HumanString(input_shape),
55 xla::ShapeUtil::HumanString(input_shape),
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| outfeed_thunk.cc | 42 if (ShapeUtil::IsEmptyTuple(hlo_instruction()->operand(0)->shape())) {
45 CHECK(ShapeUtil::Compatible(hlo_instruction()->operand(0)->shape(),
|
| while_transformer_test.cc | 33 induction_variable_shape_(ShapeUtil::MakeShape(S32, {})),
34 data_shape_(ShapeUtil::MakeShape(F32, {8})),
35 condition_result_shape_(ShapeUtil::MakeShape(PRED, {})) {}
73 ShapeUtil::MakeShape(F32, {}), induction_variable));
111 return ShapeUtil::MakeTupleShape(
114 return ShapeUtil::MakeTupleShape(
|
| instruction_fusion_test.cc | 39 ShapeUtil::MakeShape(S32, {}), HloOpcode::kExp, const0));
42 ShapeUtil::MakeShape(S32, {1}), exp1, {0}));
59 ShapeUtil::MakeShape(S32, {}), HloOpcode::kNegate, const0));
62 ShapeUtil::MakeShape(S32, {1}), negate1, {0}));
79 ShapeUtil::MakeShape(S32, {}), HloOpcode::kExp, const0));
81 HloInstruction::CreateReshape(ShapeUtil::MakeShape(S32, {}), exp1));
98 ShapeUtil::MakeShape(S32, {}), HloOpcode::kExp, const0));
100 HloInstruction::CreateTranspose(ShapeUtil::MakeShape(S32, {}), exp1, {}));
114 0, ShapeUtil::MakeShape(S32, {1, 1}), "0"));
116 CreateCanonicalDot(ShapeUtil::MakeShape(S32, {1, 1}), param0, param0))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| gpu_ftz_test.cc | 34 Shape param_shape = ShapeUtil::MakeShapeWithLayout(
51 Shape param_shape = ShapeUtil::MakeShapeWithLayout(
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| case_op.cc | 165 xla::ShapeUtil::Compatible(branch0_input_shape, branch_input_shape),
168 xla::ShapeUtil::HumanString(branch0_input_shape), " vs. ",
169 xla::ShapeUtil::HumanString(branch_input_shape)));
174 xla::ShapeUtil::Compatible(branch_results[0].xla_output_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...] |
