| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 111 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
|
| D | scatter_expander.cc | 133 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
|
| D | batch_dot_simplification.cc | 44 const DotDimensionNumbers& dim_numbers = batch_dot->dot_dimension_numbers(); in ElideDegenerateBatchDimensionFromBatchDot() local
|
| D | indexed_array_analysis.cc | 254 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()
|
| D | convolution_group_converter.cc | 201 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleBatchGroupCount() local
431 ConvolutionDimensionNumbers dim_numbers = in HandleConvolution() local
|
| D | space_to_batch_converter.cc | 278 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
|
| D | hlo_evaluator.cc | 383 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
|
| D | hlo_cost_analysis_test.cc | 977 GatherDimensionNumbers dim_numbers; in TEST_F() local
1009 ScatterDimensionNumbers dim_numbers; in TEST_F() local
|
| D | triangular_solve_expander.cc | 77 GatherDimensionNumbers dim_numbers; in DiagonalBlocks() local
|
| D | hlo_creation_utils.cc | 288 const DotDimensionNumbers& dim_numbers, in MakeDotHlo()
|
| D | shape_inference.cc | 3213 const GatherDimensionNumbers& dim_numbers) { in ValidateGatherDimensionNumbers()
3446 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in ValidateScatterDimensionNumbers()
|
| D | hlo_parser.cc | 2465 GatherDimensionNumbers dim_numbers = in ParseInstructionRhs() local
2515 ScatterDimensionNumbers dim_numbers = in ParseInstructionRhs() local
|
| D | elemental_ir_emitter.cc | 1914 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/ |
| D | scatter.cc | 138 xla::ScatterDimensionNumbers dim_numbers; in XlaScatter() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamism_inference_test.cc | 281 GatherDimensionNumbers dim_numbers; in TEST_F() local
301 GatherDimensionNumbers dim_numbers; in TEST_F() local
323 GatherDimensionNumbers dim_numbers; in TEST_F() local
|
| D | gather_operation_test.cc | 757 GatherDimensionNumbers dim_numbers; in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 122 xla::GatherDimensionNumbers dim_numbers; in XlaGather() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 482 mlir::mhlo::ScatterDimensionNumbers dim_numbers; member
|
| D | gpu_conv_rewriter.cc | 72 auto dim_numbers = conv->convolution_dimension_numbers(); in ConvertBatchGroupedToFeatureGroupedConvolution() local
|
| D | ir_emission_utils.cc | 112 const DotDimensionNumbers& dim_numbers = dot.dot_dimension_numbers(); in IsMatrixMultiplication() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 149 xla::GatherDimensionNumbers dim_numbers; in GetMatrixDiagonalViaGather() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1092 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1201 mhlo::DotDimensionNumbers dim_numbers = op.dot_dimension_numbers(); in matchAndRewrite() local
|