Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc111 HloInstruction* index_vector, const GatherDimensionNumbers& dim_numbers, in ExpandIndexVectorIntoOperandSpace()
154 const GatherDimensionNumbers& dim_numbers = gather.gather_dimension_numbers(); in GatherLoopBody() local
236 const GatherDimensionNumbers& dim_numbers) { in CreateGatherLoopAccumulatorInitValue()
277 const GatherDimensionNumbers& dim_numbers = in GatherLoopTripCount() local
334 const GatherDimensionNumbers& dim_numbers = in ExpandInstruction() local
Dscatter_expander.cc133 HloInstruction* index_vector, const ScatterDimensionNumbers& dim_numbers, in ExpandIndexVectorIntoOperandSpace()
222 const ScatterDimensionNumbers& dim_numbers = in ScatterLoopBody() local
333 const ScatterDimensionNumbers& dim_numbers = in ScatterTripCount() local
366 const ScatterDimensionNumbers& dim_numbers = in ExpandInstruction() local
Dbatch_dot_simplification.cc44 const DotDimensionNumbers& dim_numbers = batch_dot->dot_dimension_numbers(); in ElideDegenerateBatchDimensionFromBatchDot() local
Dindexed_array_analysis.cc254 const Shape& shape, const GatherDimensionNumbers& dim_numbers, in ComputeArrayForGather()
1033 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDotWithIndexedLhs()
1068 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDotWithIndexedRhs()
1103 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDot()
Dconvolution_group_converter.cc201 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleBatchGroupCount() local
431 ConvolutionDimensionNumbers dim_numbers = in HandleConvolution() local
Dspace_to_batch_converter.cc313 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in IsForwardWindowDilatedConv()
336 ConvolutionDimensionNumbers dim_numbers = in IsConvSuitableForSpaceToBatch() local
545 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers, in BringSpaceNextToBatch()
936 bool check_lhs) { in CanPropagate()
2477 ConvolutionDimensionNumbers& dim_numbers, in SplitSpace()
3047 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in GetConvolutionDetails()
3128 ConvolutionDimensionNumbers dim_numbers = in PerformSpaceToBatchOnConvolution() local
Dhlo_evaluator.cc418 const DotDimensionNumbers& dim_numbers, in EvaluateDotOp()
1515 const Shape& output_shape, const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputBatchIndices()
1534 const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputOffsetIndices()
1564 const GatherDimensionNumbers* dim_numbers, const Shape& input_shape, in OutputBatchIndexToInputIndex()
1692 const GatherDimensionNumbers& dim_numbers, const Shape& input_shape, in OutputOffsetIndexToInputIndex()
1784 const GatherDimensionNumbers& dim_numbers = in HandleGather() local
Dhlo_cost_analysis_test.cc1011 GatherDimensionNumbers dim_numbers; in TEST_F() local
1043 ScatterDimensionNumbers dim_numbers; in TEST_F() local
Dtriangular_solve_expander.cc77 GatherDimensionNumbers dim_numbers; in DiagonalBlocks() local
Dhlo_creation_utils.cc288 const DotDimensionNumbers& dim_numbers, in MakeDotHlo()
Dshape_inference.cc3350 const GatherDimensionNumbers& dim_numbers) { in ValidateGatherDimensionNumbers()
3583 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in ValidateScatterDimensionNumbers()
Delemental_ir_emitter.cc2100 const GatherDimensionNumbers& dim_numbers = hlo->gather_dimension_numbers(); in EmitElementalGather() local
2363 const DotDimensionNumbers& dim_numbers = hlo->dot_dimension_numbers(); in EmitElementalDot() local
Dhlo_parser.cc2621 GatherDimensionNumbers dim_numbers = in ParseInstructionRhs() local
2671 ScatterDimensionNumbers dim_numbers = in ParseInstructionRhs() local
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc138 xla::ScatterDimensionNumbers dim_numbers; in XlaScatter() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_einsum_to_dot_general.cc94 auto dim_numbers = mhlo::DotDimensionNumbers::get( in matchAndRewrite() local
Dlegalize_to_linalg.cc1829 mhlo::DotDimensionNumbers dim_numbers = op.dot_dimension_numbers(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op.cc122 xla::GatherDimensionNumbers dim_numbers; in XlaGather() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h412 mlir::mhlo::ScatterDimensionNumbers dim_numbers; member
Dgpu_conv_rewriter.cc72 auto dim_numbers = conv->convolution_dimension_numbers(); in ConvertBatchGroupedToFeatureGroupedConvolution() local
Dir_emission_utils.cc114 const DotDimensionNumbers& dim_numbers = dot.dot_dimension_numbers(); in IsMatrixMultiplication() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dvalue_inference_test.cc252 GatherDimensionNumbers dim_numbers; in TEST_F() local
268 GatherDimensionNumbers dim_numbers; in TEST_F() local
286 GatherDimensionNumbers dim_numbers; in TEST_F() local
Dgather_operation_test.cc757 GatherDimensionNumbers dim_numbers; in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc150 xla::GatherDimensionNumbers dim_numbers; in GetMatrixDiagonalViaGather() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc1111 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers()