| /external/tensorflow/tensorflow/compiler/xla/service/ |
| zero_sized_hlo_elimination.cc | 39 ShapeUtil::IsZeroElementArray(instruction->shape())) {
|
| hlo_constant_folding_test.cc | 48 HloInstruction::CreateConvert(ShapeUtil::MakeShape(S64, {}), input));
70 HloInstruction::CreateConvert(ShapeUtil::MakeShape(F32, {}), input));
92 HloInstruction::CreateConvert(ShapeUtil::MakeShape(S64, {2}), input));
134 Shape shape = ShapeUtil::MakeShape(F32, dimensions);
146 EXPECT_TRUE(ShapeUtil::Equal(root->shape(), shape));
158 ShapeUtil::MakeShape(F32, dimensions), 0.0, 1.0));
161 Shape shape = ShapeUtil::MakeShape(F32, {6, 6, 3, 4, 4});
173 EXPECT_TRUE(ShapeUtil::Equal(root->shape(), shape));
181 ShapeUtil::MakeShape(F32, dimensions), 0.0, 1.0));
185 Shape shape = ShapeUtil::MakeShape(F32, {8, 7, 11, 9, 5})
[all...] |
| hlo_execution_profile_test.cc | 44 Shape shape = ShapeUtil::MakeShape(F32, {30, 30});
51 return ShapeUtil::ByteSizeOf(shape, pointer_size);
|
| hlo_get_dimension_size_rewriter.cc | 40 TF_RET_CHECK(ShapeUtil::Equal(instr->shape(), legal_shape));
41 TF_RET_CHECK(ShapeUtil::HasPrimitiveType(instr->shape(), U32));
|
| hlo_graph_dumper_test.cc | 45 auto shape = ShapeUtil::MakeShape(F32, {10, 100});
129 Shape tuple_shape = ShapeUtil::MakeTupleShape(
130 {ShapeUtil::MakeShape(F32, {3, 2}), ShapeUtil::MakeShape(S32, {4, 5})});
135 ShapeUtil::MakeShape(F32, {3, 2}), constant, 0));
|
| reshape_mover.cc | 85 ShapeUtil::IsScalar(instruction->operand(0)->shape())) {
95 return !ShapeUtil::IsScalar(instruction->shape()) &&
119 !ShapeUtil::SameDimensions(a->shape(), b->shape())) {
126 return ShapeUtil::SameDimensions(a->operand(0)->shape(),
143 ShapeUtil::ChangeElementType(new_operand_shape, element_type);
172 CHECK(ShapeUtil::IsScalar(operand->operand(0)->shape()));
197 << ShapeUtil::HumanString(new_operand_shape);
204 if (ShapeUtil::IsScalar(operands[i]->shape())) {
234 ShapeUtil::ChangeElementType(new_operand_shape,
287 if (!ShapeUtil::SameDimensions(operand->shape(), instruction->shape()))
[all...] |
| hlo_instruction_test.cc | 45 Shape r0f32_ = ShapeUtil::MakeShape(F32, {});
139 EXPECT_TRUE(ShapeUtil::IsScalarWithElementType(parameter->shape(), F32));
140 EXPECT_FALSE(ShapeUtil::IsScalarWithElementType(parameter->shape(), S32));
329 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
330 Shape f32a100x10 = ShapeUtil::MakeShape(F32, {100, 10});
367 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
368 Shape f32v100 = ShapeUtil::MakeShape(F32, {100});
369 Shape f32a100x10 = ShapeUtil::MakeShape(F32, {100, 10});
718 auto shape10 = ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0});
719 auto shape01 = ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1})
[all...] |
| hlo_verifier.cc | 32 return ShapeUtil::ForEachSubshapeWithStatus(
36 ShapeUtil::HumanStringWithLayout(subshape));
205 return CheckShape(hlo, ShapeUtil::MakeShape(U32, {}));
223 if (!ShapeUtil::Equal(token->shape(), ShapeUtil::MakeTokenShape())) {
252 ShapeUtil::MakeTupleShape(
253 {infeed->infeed_shape(), ShapeUtil::MakeTokenShape()}));
269 return CheckShape(outfeed, ShapeUtil::MakeTokenShape());
275 return ShapeUtil::SameElementType(shape_0, shape_1) &&
276 (ShapeUtil::SameElementType(shape_0, result_shape) |
[all...] |
| pattern_matcher_test.cc | 69 auto scalar_shape = ShapeUtil::MakeShape(F32, {});
84 auto array_shape = ShapeUtil::MakeShape(F32, {2, 3, 4});
107 auto array_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {2, 3, 4}, 10);
130 auto tuple_shape = ShapeUtil::MakeTupleShape({
131 ShapeUtil::MakeShape(F32, {1, 2, 3}),
132 ShapeUtil::MakeShape(S32, {4, 5}),
145 ShapeUtil::Equal(*subshape, ShapeUtil::GetSubshape(tuple_shape, {0})));
149 &ShapeUtil::GetSubshape(tuple_shape, {0})))));
153 &ShapeUtil::GetSubshape(tuple_shape, {1})))))
[all...] |
| hlo_schedule_test.cc | 60 return ShapeUtil::ByteSizeOf(buffer.shape());
94 return ShapeUtil::ByteSizeOf(buffer.shape());
143 return ShapeUtil::ByteSizeOf(buffer.shape());
187 return ShapeUtil::ByteSizeOf(buffer.shape());
248 return ShapeUtil::ByteSizeOf(buffer.shape(),
259 HloInstruction::CreateUnary(ShapeUtil::MakeShape(PRED, {}),
317 return ShapeUtil::ByteSizeOf(buffer.shape(),
|
| shaped_buffer_test.cc | 38 const xla::Shape shape = xla::ShapeUtil::MakeShape(xla::F32, {});
95 Shape s = ShapeUtil::MakeShape(F32, {1});
115 Shape s = ShapeUtil::MakeShape(F32, {1});
116 s = xla::ShapeUtil::MakeTupleShape(std::vector<xla::Shape>(2, s));
117 s = xla::ShapeUtil::MakeTupleShape(std::vector<xla::Shape>(3, s));
156 xla::Shape shape = xla::ShapeUtil::MakeShape(xla::F32, {32, 64, 128});
159 shape = xla::ShapeUtil::MakeTupleShape(shapes);
|
| hlo_verifier.h | 124 return layout_sensitive_ ? ShapeUtil::Equal(a, b)
125 : ShapeUtil::Compatible(a, b);
128 return layout_sensitive_ ? ShapeUtil::EqualIgnoringFpPrecision(a, b)
129 : ShapeUtil::CompatibleIgnoringFpPrecision(a, b);
132 return layout_sensitive_ ? ShapeUtil::HumanStringWithLayout(s)
133 : ShapeUtil::HumanString(s);
139 ? ShapeUtil::SameElementTypeIgnoringFpPrecision(a, b)
140 : ShapeUtil::SameElementType(a, b);
229 [](const Shape& shape) { return ShapeUtil::ByteSizeOf(shape); })
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_layout.h | 80 string ToString() const { return ShapeUtil::HumanStringWithLayout(shape_); }
82 // Tests for equality of both shape and layout (ShapeUtil::Equal).
|
| literal_test.cc | 394 Literal nil(ShapeUtil::MakeNil());
423 Literal colmajor(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {0, 1}));
429 Literal rowmajor(ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0}));
837 Literal output(ShapeUtil::MakeShape(S64, {1}));
844 Literal output(ShapeUtil::MakeShape(U64, {2}));
851 Literal output(ShapeUtil::MakeShape(C64, {1}));
858 Literal output(ShapeUtil::MakeShape(C128, {1}));
865 Literal output(ShapeUtil::MakeShape(C64, {2, 2}));
873 Literal output(ShapeUtil::MakeShape(BF16, {}));
881 Literal output(ShapeUtil::MakeShape(BF16, {3}))
[all...] |
| literal.cc | 112 for (int i = 0; i < ShapeUtil::TupleElementCount(shape); ++i) {
131 ShapeUtil::ByteSizeOfPrimitiveType(shape.element_type())]);
227 } else if (!ShapeUtil::IsZeroElementArray(shape()) &&
228 !ShapeUtil::IsZeroElementArray(src_literal.shape())) {
261 ShapeUtil::ForEachIndex(src_literal.shape(), stride_config.base,
277 ShapeUtil::ByteSizeOfPrimitiveType(shape().element_type());
296 if (ShapeUtil::HasPrimitiveType(shape, OPAQUE)) {
303 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape));
317 ShapeUtil::TupleElementCount(piece->subshape())) {
320 ShapeUtil::TupleElementCount(piece->subshape())
[all...] |
| shape.cc | 80 return ShapeUtil::HumanStringWithLayout(*this);
82 return ShapeUtil::HumanString(*this);
137 !ShapeUtil::SameElementTypeIgnoringFpPrecision(lhs, rhs)) ||
138 (!ignore_fp_precision_ && !ShapeUtil::SameElementType(lhs, rhs))) {
164 if (!ShapeUtil::SameDimensions(lhs, rhs)) {
213 ShapeUtil::HumanString(parameters(i)));
216 ShapeUtil::HumanString(result()));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_layout_assignment_test.cc | 66 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1});
67 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 24});
68 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1});
101 Shape lhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1});
102 Shape rhs_shape = ShapeUtil::MakeShape(F32, {12, 24});
103 Shape result_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1});
143 Shape lhs_a_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 12}, {0, 1});
144 Shape lhs_b_shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 12}, {0, 1});
145 Shape rhs_shape = ShapeUtil::MakeShapeWithLayout(F32, {12, 24}, {0, 1});
146 Shape result_a_shape = ShapeUtil::MakeShapeWithLayout(F32, {1, 24}, {0, 1})
[all...] |
| xfeed_manager.cc | 34 buffer->Done(ShapeUtil::MakeNil());
73 << (shape.ok() ? ShapeUtil::HumanString(shape.ValueOrDie())
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| gpu_index_test.cc | 46 auto param_shape = ShapeUtil::MakeShape(F32, {5, 7, 2});
52 ShapeUtil::MakeShape(PRED, {5, 7, 2}), param_x, param_y,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| variadic_op_splitter_test.cc | 56 ShapeUtil::MakeShape(S32, {255}), concat_operands, 0));
71 ShapeUtil::MakeShape(S32, {256}), concat_operands, 0));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_noalias_test.cc | 45 auto param_shape = ShapeUtil::MakeShape(F32, {2, 2});
52 ShapeUtil::MakeShape(F32, {2, 4}), {param_x, param_y}, 1));
55 ShapeUtil::MakeShape(F32, {2, 6}), {concat1, param_x}, 1));
101 auto shape = ShapeUtil::MakeShape(F32, {2, 4});
111 auto shape = ShapeUtil::MakeShape(F32, {2, 6});
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| if_op.cc | 161 xla::ShapeUtil::Compatible(then_input_shape, else_input_shape),
164 xla::ShapeUtil::HumanString(then_input_shape), " vs. ",
165 xla::ShapeUtil::HumanString(else_input_shape)));
170 xla::ShapeUtil::Compatible(then_result.xla_output_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...] |
| while_op.cc | 170 if (!xla::ShapeUtil::Compatible(body_input_shape, body_output_shape)) {
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);
434 xla::ShapeUtil::Compatible(body_input_shape, cond_input_shape),
437 xla::ShapeUtil::HumanString(body_input_shape), " vs. ",
438 xla::ShapeUtil::HumanString(cond_input_shape)))
[all...] |
| xla_reduce_op.cc | 80 xla::ShapeUtil::Compatible(reducer.xla_output_shape, scalar_shape),
83 xla::ShapeUtil::HumanString(scalar_shape), " got ",
84 xla::ShapeUtil::HumanString(reducer.xla_output_shape)));
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| custom_call_test.cc | 77 Shape r0f32_ = ShapeUtil::MakeShape(F32, {});
78 Shape r2f32_ = ShapeUtil::MakeShape(F32, {2, 2});
125 ShapeUtil::MakeShape(F32, {1, 2, 2}), {input}, "Add1ToValues"));
127 ShapeUtil::MakeShape(F32, {1, 2, 2}), {incremented}, "Add1ToValues"));
131 HloInstruction::CreateConcatenate(ShapeUtil::MakeShape(F32, {2, 2, 2}),
174 ShapeUtil::MakeShapeWithLayout(F32, {2, 2}, {1, 0});
217 ShapeUtil::MakeShape(F32, {1}));
|
