| /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()
318 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
350 const ScatterDimensionNumbers& dim_numbers = in ExpandScatter() local
|
| D | batch_dot_simplification.cc | 26 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 | hlo_evaluator.cc | 364 const DotDimensionNumbers& dim_numbers, in EvaluateDotOp()
782 const Shape& output_shape, const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputBatchIndices()
801 const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputOffsetIndices()
831 const GatherDimensionNumbers* dim_numbers, const Shape& input_shape, in OutputBatchIndexToInputIndex()
958 const GatherDimensionNumbers& dim_numbers, const Shape& input_shape, in OutputOffsetIndexToInputIndex()
1050 const GatherDimensionNumbers& dim_numbers = in HandleGather() local
|
| D | convolution_group_converter.cc | 202 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleBatchGroupCount() local
334 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleConvolution() local
|
| D | hlo_evaluator_typed_visitor.h | 2059 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in IterationSpaceForUpdateScatterIndices()
2078 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in IterationSpaceForUpdateWindowIndices()
2106 const ScatterDimensionNumbers* dim_numbers, const Shape& input_shape, in UpdateScatterIndexToInputIndex()
2236 const ScatterDimensionNumbers& dim_numbers, const Shape& input_shape, in UpdateWindowIndexToInputIndex()
2308 const ScatterDimensionNumbers& dim_numbers = in HandleScatter() local
|
| D | hlo_cost_analysis_test.cc | 650 GatherDimensionNumbers dim_numbers; in TEST_F() local
677 ScatterDimensionNumbers dim_numbers; in TEST_F() local
|
| D | triangular_solve_expander.cc | 74 GatherDimensionNumbers dim_numbers; in DiagonalBlocks() local
|
| D | hlo_creation_utils.cc | 218 const DotDimensionNumbers& dim_numbers, in MakeDotHlo()
|
| D | shape_inference.cc | 2868 const GatherDimensionNumbers& dim_numbers) { in ValidateGatherDimensionNumbers()
3071 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in ValidateScatterDimensionNumbers()
|
| D | elemental_ir_emitter.cc | 1867 const GatherDimensionNumbers& dim_numbers = hlo->gather_dimension_numbers(); in EmitElementalGather() local
2118 const DotDimensionNumbers& dim_numbers = hlo->dot_dimension_numbers(); in EmitElementalDot() local
|
| D | hlo_parser.cc | 1636 GatherDimensionNumbers dim_numbers = in ParseInstructionRhs() local
1672 ScatterDimensionNumbers dim_numbers = in ParseInstructionRhs() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | scatter.cc | 138 xla::ScatterDimensionNumbers dim_numbers; in XlaScatter() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 116 xla::GatherDimensionNumbers dim_numbers; in XlaGather() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 69 const DotDimensionNumbers& dim_numbers = dot.dot_dimension_numbers(); in DotImplementedAsGemm() local
|
| D | ir_emitter_unnested.cc | 1081 const ScatterDimensionNumbers& dim_numbers = in EmitScatter() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | gather_operation_test.cc | 646 GatherDimensionNumbers dim_numbers; in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 896 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers()
|