Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc111 HloInstruction* index_vector, const GatherDimensionNumbers& dim_numbers, in ExpandIndexVectorIntoOperandSpace()
153 const GatherDimensionNumbers& dim_numbers = gather.gather_dimension_numbers(); in GatherLoopBody() local
235 const GatherDimensionNumbers& dim_numbers) { in CreateGatherLoopAccumulatorInitValue()
276 const GatherDimensionNumbers& dim_numbers = in GatherLoopTripCount() local
333 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.cc278 ConvolutionDimensionNumbers dim_numbers = in IsConvSuitableForSpaceToBatch() local
479 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers, in BringSpaceNextToBatch()
2092 ConvolutionDimensionNumbers& dim_numbers, in SplitSpace()
2592 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in GetConvolutionDetails()
2658 ConvolutionDimensionNumbers dim_numbers = in PerformSpaceToBatchOnConvolution() local
Dhlo_evaluator.cc383 const DotDimensionNumbers& dim_numbers, in EvaluateDotOp()
1451 const Shape& output_shape, const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputBatchIndices()
1470 const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputOffsetIndices()
1500 const GatherDimensionNumbers* dim_numbers, const Shape& input_shape, in OutputBatchIndexToInputIndex()
1628 const GatherDimensionNumbers& dim_numbers, const Shape& input_shape, in OutputOffsetIndexToInputIndex()
1720 const GatherDimensionNumbers& dim_numbers = in HandleGather() local
Dhlo_cost_analysis_test.cc977 GatherDimensionNumbers dim_numbers; in TEST_F() local
1009 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.cc3213 const GatherDimensionNumbers& dim_numbers) { in ValidateGatherDimensionNumbers()
3446 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in ValidateScatterDimensionNumbers()
Dhlo_parser.cc2465 GatherDimensionNumbers dim_numbers = in ParseInstructionRhs() local
2515 ScatterDimensionNumbers dim_numbers = in ParseInstructionRhs() local
Delemental_ir_emitter.cc1914 const GatherDimensionNumbers& dim_numbers = hlo->gather_dimension_numbers(); in EmitElementalGather() local
2177 const DotDimensionNumbers& dim_numbers = hlo->dot_dimension_numbers(); in EmitElementalDot() local
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc138 xla::ScatterDimensionNumbers dim_numbers; in XlaScatter() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamism_inference_test.cc281 GatherDimensionNumbers dim_numbers; in TEST_F() local
301 GatherDimensionNumbers dim_numbers; in TEST_F() local
323 GatherDimensionNumbers dim_numbers; in TEST_F() local
Dgather_operation_test.cc757 GatherDimensionNumbers dim_numbers; in XLA_TEST_F() 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.h482 mlir::mhlo::ScatterDimensionNumbers dim_numbers; member
Dgpu_conv_rewriter.cc72 auto dim_numbers = conv->convolution_dimension_numbers(); in ConvertBatchGroupedToFeatureGroupedConvolution() local
Dir_emission_utils.cc112 const DotDimensionNumbers& dim_numbers = dot.dot_dimension_numbers(); in IsMatrixMultiplication() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc149 xla::GatherDimensionNumbers dim_numbers; in GetMatrixDiagonalViaGather() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc1092 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc1201 mhlo::DotDimensionNumbers dim_numbers = op.dot_dimension_numbers(); in matchAndRewrite() local