Home
last modified time | relevance | path

Searched defs:dim_numbers (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dgather_expander.cc113 HloInstruction* index_vector, const GatherDimensionNumbers& dim_numbers, in ExpandIndexVectorIntoOperandSpace()
156 const GatherDimensionNumbers& dim_numbers = gather.gather_dimension_numbers(); in GatherLoopBody() local
237 const GatherDimensionNumbers& dim_numbers) { in CreateGatherLoopAccumulatorInitValue()
278 const GatherDimensionNumbers& dim_numbers = in GatherLoopTripCount() local
359 const GatherDimensionNumbers& dim_numbers = in ExpandInstruction() local
H A Dscatter_expander.cc136 HloInstruction* index_vector, const ScatterDimensionNumbers& dim_numbers, in ExpandIndexVectorIntoOperandSpace()
251 const ScatterDimensionNumbers& dim_numbers = in ScatterLoopBody() local
384 const ScatterDimensionNumbers& dim_numbers = in ScatterTripCount() local
418 const ScatterDimensionNumbers& dim_numbers = in ExpandInstruction() local
H A Dhlo_evaluator.cc953 const DotDimensionNumbers& dim_numbers, in EvaluateDotOp()
2197 const Shape& output_shape, const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputBatchIndices()
2216 const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputOffsetIndices()
2246 const GatherDimensionNumbers* dim_numbers, const Shape& input_shape, in OutputBatchIndexToInputIndex()
2374 const GatherDimensionNumbers& dim_numbers, const Shape& input_shape, in OutputOffsetIndexToInputIndex()
2466 const GatherDimensionNumbers& dim_numbers = in HandleGather() local
2580 const ScatterDimensionNumbers& dim_numbers) { in GetIterationSpaceImpl()
2609 const ScatterDimensionNumbers& dim_numbers) { in IterationSpaceForUpdateScatterIndices()
2619 const ScatterDimensionNumbers& dim_numbers) { in IterationSpaceForUpdateWindowIndices()
2637 const ScatterDimensionNumbers& dim_numbers, int64_t input_rank, in UpdateScatterIndexToInputIndex()
[all …]
H A Dindexed_array_analysis.cc259 const Shape& shape, const GatherDimensionNumbers& dim_numbers, in ComputeArrayForGather()
1037 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDotWithIndexedLhs()
1072 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDotWithIndexedRhs()
1107 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDot()
H A Dscatter_simplifier.cc114 ScatterDimensionNumbers dim_numbers; in MakeScatterDimensionNumbers() local
163 auto dim_numbers = MakeScatterDimensionNumbers( in ExpandInstruction() local
H A Dspace_to_batch_converter.cc373 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in IsForwardWindowDilatedConv()
396 ConvolutionDimensionNumbers dim_numbers = in IsConvSuitableForSpaceToBatch() local
480 auto dim_numbers = convolution->convolution_dimension_numbers(); in IsThisBackPropFilterConv() local
640 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers, in BringSpaceNextToBatch()
1080 std::vector<int64_t>& dim_map, bool check_lhs) { in CanPropagate()
2936 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers, in SplitSpace()
3543 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in GetConvolutionDetails()
3636 ConvolutionDimensionNumbers dim_numbers = in PerformSpaceToBatchOnConvolution() local
H A Dconvolution_group_converter.cc205 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleBatchGroupCount() local
436 ConvolutionDimensionNumbers dim_numbers = in HandleConvolution() local
H A Dhlo_cost_analysis_test.cc1148 GatherDimensionNumbers dim_numbers; in TEST_F() local
1180 ScatterDimensionNumbers dim_numbers; in TEST_F() local
1217 ScatterDimensionNumbers dim_numbers; in TEST_F() local
H A Dbatch_dot_simplification.cc43 const DotDimensionNumbers& dim_numbers = batch_dot->dot_dimension_numbers(); in ElideDegenerateBatchDimensionFromBatchDot() local
H A Dtriangular_solve_expander.cc72 GatherDimensionNumbers dim_numbers; in DiagonalBlocks() local
H A Dshape_inference.cc3332 const GatherDimensionNumbers& dim_numbers) { in ValidateGatherDimensionNumbers()
3565 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in ValidateScatterDimensionNumbers()
H A Dhlo_creation_utils.cc336 const DotDimensionNumbers& dim_numbers, in MakeDotHlo()
H A Delemental_ir_emitter.cc2105 const GatherDimensionNumbers& dim_numbers = hlo->gather_dimension_numbers(); in EmitElementalGather() local
2370 const DotDimensionNumbers& dim_numbers = hlo->dot_dimension_numbers(); in EmitElementalDot() local
H A Dhlo_parser.cc2829 GatherDimensionNumbers dim_numbers = in CreateInstruction() local
2882 ScatterDimensionNumbers dim_numbers = in CreateInstruction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dvalue_inference_test.cc278 GatherDimensionNumbers dim_numbers; in TEST_F() local
294 GatherDimensionNumbers dim_numbers; in TEST_F() local
312 GatherDimensionNumbers dim_numbers; in TEST_F() local
H A Dgather_operation_test.cc757 GatherDimensionNumbers dim_numbers; in XLA_TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/lib/
H A Dscatter.cc138 xla::ScatterDimensionNumbers dim_numbers; in XlaScatter() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.h485 mlir::mhlo::ScatterDimensionNumbersAttr dim_numbers; member
H A Dgpu_conv_rewriter.cc608 auto dim_numbers = conv->convolution_dimension_numbers(); in ConvertBatchGroupedToFeatureGroupedConvolution() local
H A Dir_emission_utils.cc101 const DotDimensionNumbers& dim_numbers = dot.dot_dimension_numbers(); in IsMatrixMultiplication() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dgather_op.cc123 xla::GatherDimensionNumbers dim_numbers; in XlaGather() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dmatrix.cc150 xla::GatherDimensionNumbers dim_numbers; in GetMatrixDiagonalViaGather() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Ddot_op_emitter.cc1289 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers()