Home
last modified time | relevance | path

Searched refs:IsZeroElementArray (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc360 TEST(ShapeUtilTest, IsZeroElementArray) { in TEST() argument
361 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {}))); in TEST()
362 EXPECT_TRUE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0}))); in TEST()
363 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1}))); in TEST()
365 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1, 1}))); in TEST()
366 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2}))); in TEST()
368 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2, 1}))); in TEST()
370 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 5}))); in TEST()
372 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 0, 5}))); in TEST()
374 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0, 3, 0}))); in TEST()
[all …]
Dshape_util.h199 static bool IsZeroElementArray(const Shape& shape);
728 if (ShapeUtil::IsZeroElementArray(shape)) {
Dliteral.cc227 } else if (!ShapeUtil::IsZeroElementArray(shape()) && in CopySliceFromInternal()
228 !ShapeUtil::IsZeroElementArray(src_literal.shape())) { in CopySliceFromInternal()
375 if (ShapeUtil::IsZeroElementArray(dest_shape)) { in CopyElementsBetween()
1223 if (ShapeUtil::IsZeroElementArray(shape())) { in EachCellAsString()
1669 if (ShapeUtil::IsZeroElementArray(piece.subshape())) { in IsAllFirst()
Dliteral.h981 if (ShapeUtil::IsZeroElementArray(shape())) { in EachCell()
Dshape_util.cc414 /* static */ bool ShapeUtil::IsZeroElementArray(const Shape& shape) { in IsZeroElementArray() function in xla::ShapeUtil
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc67 if (ShapeUtil::IsZeroElementArray(input_shape) || in PotentiallyImplementedAsEigenConvolution()
68 ShapeUtil::IsZeroElementArray(kernel_shape)) { in PotentiallyImplementedAsEigenConvolution()
Ddot_op_emitter.cc773 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) || in IsAlignedGemm()
774 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) { in IsAlignedGemm()
Dir_emitter.cc1934 if (ShapeUtil::IsZeroElementArray(slice->shape())) { in HandleSlice()
/external/tensorflow/tensorflow/compiler/xla/service/
Dzero_sized_hlo_elimination.cc39 ShapeUtil::IsZeroElementArray(instruction->shape())) { in Run()
Dgather_expander.cc309 CHECK(!ShapeUtil::IsZeroElementArray(gather_instr->shape())); in ExpandInstruction()
376 !ShapeUtil::IsZeroElementArray(inst->shape()); in InstructionMatchesPattern()
Dalgebraic_simplifier.cc655 if (!ShapeUtil::IsZeroElementArray(operand->shape())) { in HandleConcatenate()
1638 if (ShapeUtil::IsZeroElementArray(dot->shape()) || in HandleDot()
1639 ShapeUtil::IsZeroElementArray(lhs->shape()) || in HandleDot()
1640 ShapeUtil::IsZeroElementArray(rhs->shape())) { in HandleDot()
2195 if (ShapeUtil::IsZeroElementArray(pad->operand(0)->shape())) { in HandlePad()
2593 if (ShapeUtil::IsZeroElementArray(reshape->shape())) { in HandleReshape()
2901 if (ShapeUtil::IsZeroElementArray(update->shape())) { in HandleDynamicUpdateSlice()
2922 if (ShapeUtil::IsZeroElementArray(arg->shape()) || in HandleReduce()
2923 ShapeUtil::IsZeroElementArray(reduce_result_shape)) { in HandleReduce()
3085 if (ShapeUtil::IsZeroElementArray(reduce_window->operand(0)->shape())) { in HandleReduceWindow()
[all …]
Dscatter_expander.cc355 if (ShapeUtil::IsZeroElementArray(updates->shape())) { in ExpandScatter()
Dtriangular_solve_expander.cc409 if (ShapeUtil::IsZeroElementArray(b_shape)) { in BuildTriangularSolve()
Dhlo_graph_dumper.cc823 if (ShapeUtil::IsZeroElementArray(shape)) { in GetInstructionNodeInlinedOperands()
Dshape_inference.cc1859 if (ShapeUtil::IsZeroElementArray(in)) { in InferFftShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc61 !ShapeUtil::IsZeroElementArray(lhs_shape) && in AreValidGemmShapes()
62 !ShapeUtil::IsZeroElementArray(rhs_shape); in AreValidGemmShapes()
Dcudnn_conv_rewriter.cc75 if (ShapeUtil::IsZeroElementArray(conv->operand(0)->shape()) || in CanImplementAsCudnnForwardConv()
76 ShapeUtil::IsZeroElementArray(conv->operand(1)->shape())) { in CanImplementAsCudnnForwardConv()
Dir_emitter.cc638 if (ShapeUtil::IsZeroElementArray(convolution->shape())) { in HandleConvolution()
648 if (ShapeUtil::IsZeroElementArray(fft->shape())) { in HandleFft()