| /external/tensorflow/tensorflow/compiler/xla/service/ |
| zero_sized_hlo_elimination.cc | 39 ShapeUtil::IsZeroElementArray(instruction->shape())) {
|
| gather_expander.cc | 309 CHECK(!ShapeUtil::IsZeroElementArray(gather_instr->shape()));
376 !ShapeUtil::IsZeroElementArray(inst->shape());
|
| algebraic_simplifier.cc | 655 if (!ShapeUtil::IsZeroElementArray(operand->shape())) {
[all...] |
| scatter_expander.cc | 355 if (ShapeUtil::IsZeroElementArray(updates->shape())) {
|
| triangular_solve_expander.cc | 409 if (ShapeUtil::IsZeroElementArray(b_shape)) {
|
| hlo_graph_dumper.cc | 823 if (ShapeUtil::IsZeroElementArray(shape)) {
[all...] |
| shape_inference.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emission_utils.cc | 67 if (ShapeUtil::IsZeroElementArray(input_shape) ||
68 ShapeUtil::IsZeroElementArray(kernel_shape)) {
|
| dot_op_emitter.cc | 773 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) ||
774 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) {
[all...] |
| ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util_test.cc | 360 TEST(ShapeUtilTest, IsZeroElementArray) {
361 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {})));
362 EXPECT_TRUE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0})));
363 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1})));
365 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1, 1})));
366 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2})));
368 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2, 1})));
370 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 5})));
372 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 0, 5})));
374 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0, 3, 0})))
[all...] |
| shape_util.h | 199 static bool IsZeroElementArray(const Shape& shape);
728 if (ShapeUtil::IsZeroElementArray(shape)) {
[all...] |
| literal.cc | 227 } else if (!ShapeUtil::IsZeroElementArray(shape()) &&
228 !ShapeUtil::IsZeroElementArray(src_literal.shape())) {
375 if (ShapeUtil::IsZeroElementArray(dest_shape)) {
[all...] |
| literal.h | [all...] |
| shape_util.cc | 414 /* static */ bool ShapeUtil::IsZeroElementArray(const Shape& shape) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| ir_emission_utils.cc | 61 !ShapeUtil::IsZeroElementArray(lhs_shape) &&
62 !ShapeUtil::IsZeroElementArray(rhs_shape);
|
| cudnn_conv_rewriter.cc | 75 if (ShapeUtil::IsZeroElementArray(conv->operand(0)->shape()) ||
76 ShapeUtil::IsZeroElementArray(conv->operand(1)->shape())) {
|
| ir_emitter.cc | 638 if (ShapeUtil::IsZeroElementArray(convolution->shape())) {
648 if (ShapeUtil::IsZeroElementArray(fft->shape())) {
[all...] |
