| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.h | 33 mlir::mhlo::GatherDimensionNumbers ConvertGatherDimensionNumbers(
34 const xla::GatherDimensionNumbers& dnums, mlir::Builder* builder);
|
| D | hlo_utils.h | 47 mlir::mhlo::GatherDimensionNumbers CreateGatherDimensionNumbers(
48 const GatherDimensionNumbers& input, mlir::Builder builder);
|
| D | attribute_importer.cc | 48 mlir::mhlo::GatherDimensionNumbers ConvertGatherDimensionNumbers( in ConvertGatherDimensionNumbers()
49 const xla::GatherDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertGatherDimensionNumbers()
56 return mlir::mhlo::GatherDimensionNumbers::get( in ConvertGatherDimensionNumbers()
|
| D | hlo_utils.cc | 261 mlir::mhlo::GatherDimensionNumbers CreateGatherDimensionNumbers( in CreateGatherDimensionNumbers()
262 const GatherDimensionNumbers& input, mlir::Builder builder) { in CreateGatherDimensionNumbers()
279 return mlir::mhlo::GatherDimensionNumbers::get( in CreateGatherDimensionNumbers()
|
| D | mlir_hlo_to_hlo.cc | 293 static xla::GatherDimensionNumbers Convert_dimension_numbers( in Convert_dimension_numbers()
294 mlir::mhlo::GatherDimensionNumbers input) { in Convert_dimension_numbers()
295 xla::GatherDimensionNumbers output; in Convert_dimension_numbers()
|
| /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()
235 const GatherDimensionNumbers& dim_numbers) { in CreateGatherLoopAccumulatorInitValue()
276 const GatherDimensionNumbers& dim_numbers = in GatherLoopTripCount()
333 const GatherDimensionNumbers& dim_numbers = in ExpandInstruction()
|
| D | indexed_array_analysis.h | 265 const Shape& shape, const GatherDimensionNumbers& dim_numbers,
|
| D | hlo_instructions.h | 1657 const GatherDimensionNumbers& gather_dim_numbers,
1659 const GatherDimensionNumbers& gather_dimension_numbers() const { in gather_dimension_numbers()
1674 static GatherDimensionNumbers MakeGatherDimNumbers(
1680 const GatherDimensionNumbers& gather_dimension_numbers);
1693 std::unique_ptr<GatherDimensionNumbers> gather_dimension_numbers_;
|
| D | shape_inference.h | 315 const GatherDimensionNumbers& gather_dim_numbers,
|
| D | hlo_evaluator.cc | 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()
1616 const GatherDimensionNumbers& dim_numbers_;
1628 const GatherDimensionNumbers& dim_numbers, const Shape& input_shape, in OutputOffsetIndexToInputIndex()
1720 const GatherDimensionNumbers& dim_numbers = in HandleGather()
|
| D | hlo_sharding_util.cc | 505 const GatherDimensionNumbers& dnums = hlo->gather_dimension_numbers(); in GatherOutputSharding()
544 const GatherDimensionNumbers& dnums = hlo->gather_dimension_numbers(); in GatherIndexSharding()
593 const GatherDimensionNumbers& dnums = hlo.gather_dimension_numbers(); in GatherEffectiveOutputSharding()
|
| D | hlo_instructions.cc | 2725 const GatherDimensionNumbers& gather_dim_numbers, in HloGatherInstruction()
2732 absl::make_unique<GatherDimensionNumbers>(gather_dim_numbers); in HloGatherInstruction()
2737 const GatherDimensionNumbers& gather_dimension_numbers) { in GatherDimensionNumbersToString()
2755 /* static */ GatherDimensionNumbers HloGatherInstruction::MakeGatherDimNumbers( in MakeGatherDimNumbers()
2759 GatherDimensionNumbers gather_dim_numbers; in MakeGatherDimNumbers()
|
| D | hlo_instruction.h | 946 const GatherDimensionNumbers& gather_dim_numbers,
1883 const GatherDimensionNumbers& gather_dimension_numbers() const;
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base_structs.td | 55 def GatherDimensionNumbers : StructAttr<"GatherDimensionNumbers", HLO_Dialect,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamism_inference_test.cc | 281 GatherDimensionNumbers dim_numbers; in TEST_F()
301 GatherDimensionNumbers dim_numbers; in TEST_F()
323 GatherDimensionNumbers dim_numbers; in TEST_F()
|
| D | gather_operation_test.cc | 757 GatherDimensionNumbers dim_numbers; in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_scatter_ops.cc | 48 xla::GatherDimensionNumbers dnums_;
|
| D | reverse_sequence_op.cc | 100 xla::GatherDimensionNumbers dnums; in Compile()
|
| D | gather_op.cc | 122 xla::GatherDimensionNumbers dim_numbers; in XlaGather()
|
| D | resampler_ops.cc | 140 xla::GatherDimensionNumbers gather_dim_numbers; in Gather2by2Neighbors()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 139 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in PartitionIndexOnlyPartition()
185 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in ParititonPassthroughOperand()
220 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in ParititonTrivialIndexedOperandDimension()
327 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in PartitionIndexParallelDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 216 GatherDimensionNumbers gather_dnums; in TorchGather()
279 GatherDimensionNumbers gather_dnums; in TorchIndexSelect()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | types.h | 290 struct type_caster<xla::GatherDimensionNumbers> {
292 PYBIND11_TYPE_CASTER(xla::GatherDimensionNumbers,
|
| D | xla_client.py | 639 class GatherDimensionNumbers(object): class
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 181 const GatherDimensionNumbers& dimension_numbers,
|