Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc384 TEST(ShapeUtilTest, IsZeroElementArray) { in TEST() argument
385 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {}))); in TEST()
386 EXPECT_TRUE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0}))); in TEST()
387 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1}))); in TEST()
389 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1, 1}))); in TEST()
390 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2}))); in TEST()
392 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2, 1}))); in TEST()
394 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 5}))); in TEST()
396 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 0, 5}))); in TEST()
398 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0, 3, 0}))); in TEST()
[all …]
Dshape_util.h208 static bool IsZeroElementArray(const Shape& shape);
840 if (ShapeUtil::IsZeroElementArray(shape)) {
Dliteral.cc287 } else if (!ShapeUtil::IsZeroElementArray(shape()) && in CopySliceFromInternal()
288 !ShapeUtil::IsZeroElementArray(src_literal.shape())) { in CopySliceFromInternal()
441 if (ShapeUtil::IsZeroElementArray(dest_shape)) { in CopyElementsBetween()
1328 if (ShapeUtil::IsZeroElementArray(shape())) { in EachCellAsString()
1574 if (ShapeUtil::IsZeroElementArray(dest_shape)) { in CopyElementsWithDynamicBound()
1837 if (ShapeUtil::IsZeroElementArray(piece.subshape())) { in IsAllFirst()
Dliteral.h978 if (ShapeUtil::IsZeroElementArray(shape())) { in EachCell()
Dshape_util.cc571 /* 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.cc963 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) || in IsAlignedGemm()
964 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) { in IsAlignedGemm()
Dir_emitter.cc1848 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.cc325 CHECK(!ShapeUtil::IsZeroElementArray(gather_instr->shape())); in ExpandInstruction()
385 !ShapeUtil::IsZeroElementArray(inst->shape()) && in InstructionMatchesPattern()
Dalgebraic_simplifier.cc1219 if (!ShapeUtil::IsZeroElementArray(operand->shape())) { in HandleConcatenate()
2313 if (ShapeUtil::IsZeroElementArray(dot->shape()) || in HandleDot()
2314 ShapeUtil::IsZeroElementArray(lhs->shape()) || in HandleDot()
2315 ShapeUtil::IsZeroElementArray(rhs->shape())) { in HandleDot()
2511 if (ShapeUtil::IsZeroElementArray(operand_shape)) { in HandleGather()
3323 if (ShapeUtil::IsZeroElementArray(pad->operand(0)->shape())) { in HandlePad()
3877 if (ShapeUtil::IsZeroElementArray(reshape->shape())) { in HandleReshape()
4506 if (ShapeUtil::IsZeroElementArray(dus_update->shape())) { in HandleDynamicUpdateSlice()
4525 if (ShapeUtil::IsZeroElementArray(arg->shape()) || in HandleReduce()
4526 ShapeUtil::IsZeroElementArray(reduce_result_shape)) { in HandleReduce()
[all …]
Dscatter_expander.cc371 if (ShapeUtil::IsZeroElementArray(updates->shape())) { in ExpandInstruction()
Dtriangular_solve_expander.cc527 if (ShapeUtil::IsZeroElementArray(b_shape)) { in BuildTriangularSolve()
Dhlo_graph_dumper.cc835 if (ShapeUtil::IsZeroElementArray(shape)) { in GetInstructionNodeInlinedOperands()
Dhlo_evaluator.cc1754 if (ShapeUtil::IsZeroElementArray(operand_shape)) { in HandleGather()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter.cc129 if (ShapeUtil::IsZeroElementArray(conv->operand(0)->shape()) || in CanImplementAsGpuForwardConv()
130 ShapeUtil::IsZeroElementArray(conv->operand(1)->shape())) { in CanImplementAsGpuForwardConv()
Dir_emitter.cc550 if (ShapeUtil::IsZeroElementArray(convolution->shape())) { in HandleConvolution()
560 if (ShapeUtil::IsZeroElementArray(fft->shape())) { in HandleFft()
Dir_emission_utils.cc69 !ShapeUtil::IsZeroElementArray(lhs_shape) && in AreValidGemmShapes()
70 !ShapeUtil::IsZeroElementArray(rhs_shape); in AreValidGemmShapes()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc219 if (!xla::ShapeUtil::IsZeroElementArray(expected)) { in BuildComputationInputs()