Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util_test.cc399 TEST(ShapeUtilTest, IsZeroElementArray) { in TEST() argument
400 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {}))); in TEST()
401 EXPECT_TRUE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0}))); in TEST()
402 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1}))); in TEST()
404 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {1, 1}))); in TEST()
405 EXPECT_FALSE(ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2}))); in TEST()
407 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {2, 1}))); in TEST()
409 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 5}))); in TEST()
411 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {3, 0, 5}))); in TEST()
413 ShapeUtil::IsZeroElementArray(ShapeUtil::MakeShape(S32, {0, 3, 0}))); in TEST()
[all …]
H A Dshape_util.h118 static bool IsZeroElementArray(const Shape& shape);
779 if (ShapeUtil::IsZeroElementArray(shape)) {
H A Dliteral.cc392 } else if (!ShapeUtil::IsZeroElementArray(shape()) && in CopySliceFromInternal()
393 !ShapeUtil::IsZeroElementArray(src_literal.shape())) { in CopySliceFromInternal()
559 if (ShapeUtil::IsZeroElementArray(dest_shape)) { in CopyElementsBetween()
1479 if (ShapeUtil::IsZeroElementArray(shape())) { in EachCellAsString()
1773 if (ShapeUtil::IsZeroElementArray(dest_shape)) { in CopyElementsWithDynamicBound()
2097 if (ShapeUtil::IsZeroElementArray(shape())) { in IsAllFirst()
H A Dliteral.h1279 if (ShapeUtil::IsZeroElementArray(shape())) { in EachCell()
1297 if (ShapeUtil::IsZeroElementArray(shape())) { in MutableEachCell()
H A Dshape_util.cc633 /* static */ bool ShapeUtil::IsZeroElementArray(const Shape& shape) { in IsZeroElementArray() function in xla::ShapeUtil
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emission_utils.cc66 if (ShapeUtil::IsZeroElementArray(input_shape) || in PotentiallyImplementedAsEigenConvolution()
67 ShapeUtil::IsZeroElementArray(kernel_shape)) { in PotentiallyImplementedAsEigenConvolution()
H A Ddot_op_emitter.cc1146 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) || in IsAlignedGemm()
1147 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) { in IsAlignedGemm()
H A Dir_emitter.cc1959 if (ShapeUtil::IsZeroElementArray(slice->shape())) { in HandleSlice()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dgather_expander.cc332 CHECK(!ShapeUtil::IsZeroElementArray(gather_instr->shape())); in ExpandInstruction()
335 if (ShapeUtil::IsZeroElementArray(gather_instr->operand(0)->shape())) { in ExpandInstruction()
410 !ShapeUtil::IsZeroElementArray(inst->shape()) && in InstructionMatchesPattern()
H A Dzero_sized_hlo_elimination.cc42 ShapeUtil::IsZeroElementArray(instruction->shape()) && in Run()
H A Dalgebraic_simplifier.cc1371 if (!ShapeUtil::IsZeroElementArray(operand->shape())) { in HandleConcatenate()
2622 if (ShapeUtil::IsZeroElementArray(dot->shape()) || in HandleDot()
2623 ShapeUtil::IsZeroElementArray(lhs->shape()) || in HandleDot()
2624 ShapeUtil::IsZeroElementArray(rhs->shape())) { in HandleDot()
2788 if (ShapeUtil::IsZeroElementArray(operand_shape)) { in HandleGather()
3702 if (ShapeUtil::IsZeroElementArray(pad->operand(0)->shape())) { in HandlePad()
4253 if (ShapeUtil::IsZeroElementArray(reshape->shape())) { in HandleReshape()
5122 if (ShapeUtil::IsZeroElementArray(dus_update->shape())) { in HandleDynamicUpdateSlice()
5292 if (ShapeUtil::IsZeroElementArray(arg->shape()) || in HandleReduce()
5293 ShapeUtil::IsZeroElementArray(reduce_result_shape)) { in HandleReduce()
[all …]
H A Dscatter_expander.cc423 if (ShapeUtil::IsZeroElementArray(scatter_updates[0]->shape())) { in ExpandInstruction()
H A Dtriangular_solve_expander.cc520 if (ShapeUtil::IsZeroElementArray(b_shape)) { in BuildTriangularSolve()
H A Dhlo_graph_dumper.cc873 if (ShapeUtil::IsZeroElementArray(shape)) { in GetInstructionNodeInlinedOperands()
H A Dhlo_evaluator.cc2500 if (ShapeUtil::IsZeroElementArray(operand_shape)) { in HandleGather()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter.cc532 if (ShapeUtil::IsZeroElementArray(convolution->shape())) { in HandleConvolution()
542 if (ShapeUtil::IsZeroElementArray(fft->shape())) { in HandleFft()
H A Dgpu_conv_rewriter.cc50 if (ShapeUtil::IsZeroElementArray(conv->operand(0)->shape()) || in CanImplementAsGpuForwardConv()
51 ShapeUtil::IsZeroElementArray(conv->operand(1)->shape())) { in CanImplementAsGpuForwardConv()
H A Dir_emission_utils.cc115 !ShapeUtil::IsZeroElementArray(lhs_shape) && in IsMatrixMultiplication()
116 !ShapeUtil::IsZeroElementArray(rhs_shape); in IsMatrixMultiplication()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_reshard_variables_op_util.cc121 if (!xla::ShapeUtil::IsZeroElementArray(expected)) { in BuildInputBuffers()
H A Dtpu_execute_op.cc218 if (!xla::ShapeUtil::IsZeroElementArray(expected)) { in BuildComputationInputs()