/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
H A D | shape_util_test.cc | 399 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 D | shape_util.h | 118 static bool IsZeroElementArray(const Shape& shape); 779 if (ShapeUtil::IsZeroElementArray(shape)) {
|
H A D | literal.cc | 392 } 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 D | literal.h | 1279 if (ShapeUtil::IsZeroElementArray(shape())) { in EachCell() 1297 if (ShapeUtil::IsZeroElementArray(shape())) { in MutableEachCell()
|
H A D | shape_util.cc | 633 /* 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 D | ir_emission_utils.cc | 66 if (ShapeUtil::IsZeroElementArray(input_shape) || in PotentiallyImplementedAsEigenConvolution() 67 ShapeUtil::IsZeroElementArray(kernel_shape)) { in PotentiallyImplementedAsEigenConvolution()
|
H A D | dot_op_emitter.cc | 1146 if (ShapeUtil::IsZeroElementArray(dot_info.lhs_shape) || in IsAlignedGemm() 1147 ShapeUtil::IsZeroElementArray(dot_info.rhs_shape)) { in IsAlignedGemm()
|
H A D | ir_emitter.cc | 1959 if (ShapeUtil::IsZeroElementArray(slice->shape())) { in HandleSlice()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | gather_expander.cc | 332 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 D | zero_sized_hlo_elimination.cc | 42 ShapeUtil::IsZeroElementArray(instruction->shape()) && in Run()
|
H A D | algebraic_simplifier.cc | 1371 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 D | scatter_expander.cc | 423 if (ShapeUtil::IsZeroElementArray(scatter_updates[0]->shape())) { in ExpandInstruction()
|
H A D | triangular_solve_expander.cc | 520 if (ShapeUtil::IsZeroElementArray(b_shape)) { in BuildTriangularSolve()
|
H A D | hlo_graph_dumper.cc | 873 if (ShapeUtil::IsZeroElementArray(shape)) { in GetInstructionNodeInlinedOperands()
|
H A D | hlo_evaluator.cc | 2500 if (ShapeUtil::IsZeroElementArray(operand_shape)) { in HandleGather()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | ir_emitter.cc | 532 if (ShapeUtil::IsZeroElementArray(convolution->shape())) { in HandleConvolution() 542 if (ShapeUtil::IsZeroElementArray(fft->shape())) { in HandleFft()
|
H A D | gpu_conv_rewriter.cc | 50 if (ShapeUtil::IsZeroElementArray(conv->operand(0)->shape()) || in CanImplementAsGpuForwardConv() 51 ShapeUtil::IsZeroElementArray(conv->operand(1)->shape())) { in CanImplementAsGpuForwardConv()
|
H A D | ir_emission_utils.cc | 115 !ShapeUtil::IsZeroElementArray(lhs_shape) && in IsMatrixMultiplication() 116 !ShapeUtil::IsZeroElementArray(rhs_shape); in IsMatrixMultiplication()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
H A D | tpu_reshard_variables_op_util.cc | 121 if (!xla::ShapeUtil::IsZeroElementArray(expected)) { in BuildInputBuffers()
|
H A D | tpu_execute_op.cc | 218 if (!xla::ShapeUtil::IsZeroElementArray(expected)) { in BuildComputationInputs()
|