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.cc382 TEST(ShapeUtilTest, IsZeroElementArray) { in TEST() argument
383 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {}))); in TEST()
384 EXPECT_TRUE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0}))); in TEST()
385 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1}))); in TEST()
387 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1, 1}))); in TEST()
388 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2}))); in TEST()
390 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2, 1}))); in TEST()
392 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 5}))); in TEST()
394 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 0, 5}))); in TEST()
396 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0, 3, 0}))); in TEST()
[all …]
Dshape_util.h207 static bool IsZeroElementArray(const Shape& shape);
770 if (ShapeUtil::IsZeroElementArray(shape)) {
Dliteral.cc219 } else if (!ShapeUtil::IsZeroElementArray(shape()) && in CopySliceFromInternal()
220 !ShapeUtil::IsZeroElementArray(src_literal.shape())) { in CopySliceFromInternal()
371 if (ShapeUtil::IsZeroElementArray(dest_shape)) { in CopyElementsBetween()
1110 if (ShapeUtil::IsZeroElementArray(shape())) { in EachCellAsString()
1556 if (ShapeUtil::IsZeroElementArray(piece.subshape())) { in IsAllFirst()
Dliteral.h915 if (ShapeUtil::IsZeroElementArray(shape())) { in EachCell()
Dshape_util.cc461 /* 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.cc824 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) || in IsAlignedGemm()
825 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) { in IsAlignedGemm()
Dir_emitter.cc2177 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.cc810 if (!ShapeUtil::IsZeroElementArray(operand->shape())) { in HandleConcatenate()
1857 if (ShapeUtil::IsZeroElementArray(dot->shape()) || in HandleDot()
1858 ShapeUtil::IsZeroElementArray(lhs->shape()) || in HandleDot()
1859 ShapeUtil::IsZeroElementArray(rhs->shape())) { in HandleDot()
2044 if (ShapeUtil::IsZeroElementArray(operand_shape)) { in HandleGather()
2678 if (ShapeUtil::IsZeroElementArray(pad->operand(0)->shape())) { in HandlePad()
3122 if (ShapeUtil::IsZeroElementArray(reshape->shape())) { in HandleReshape()
3543 if (ShapeUtil::IsZeroElementArray(update->shape())) { in HandleDynamicUpdateSlice()
3564 if (ShapeUtil::IsZeroElementArray(arg->shape()) || in HandleReduce()
3565 ShapeUtil::IsZeroElementArray(reduce_result_shape)) { in HandleReduce()
[all …]
Dscatter_expander.cc355 if (ShapeUtil::IsZeroElementArray(updates->shape())) { in ExpandScatter()
Dtriangular_solve_expander.cc426 if (ShapeUtil::IsZeroElementArray(b_shape)) { in BuildTriangularSolve()
Dhlo_graph_dumper.cc832 if (ShapeUtil::IsZeroElementArray(shape)) { in GetInstructionNodeInlinedOperands()
Dhlo_evaluator.cc1742 if (ShapeUtil::IsZeroElementArray(operand_shape)) { in HandleGather()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc64 !ShapeUtil::IsZeroElementArray(lhs_shape) && in AreValidGemmShapes()
65 !ShapeUtil::IsZeroElementArray(rhs_shape); in AreValidGemmShapes()
Dgpu_conv_rewriter.cc75 if (ShapeUtil::IsZeroElementArray(conv->operand(0)->shape()) || in CanImplementAsGpuForwardConv()
76 ShapeUtil::IsZeroElementArray(conv->operand(1)->shape())) { in CanImplementAsGpuForwardConv()
Dir_emitter.cc675 if (ShapeUtil::IsZeroElementArray(convolution->shape())) { in HandleConvolution()
685 if (ShapeUtil::IsZeroElementArray(fft->shape())) { in HandleFft()