| /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()
318 const GatherDimensionNumbers& dim_numbers = in ExpandInstruction()
|
| D | hlo_evaluator.cc | 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()
946 const GatherDimensionNumbers& dim_numbers_;
958 const GatherDimensionNumbers& dim_numbers, const Shape& input_shape, in OutputOffsetIndexToInputIndex()
1050 const GatherDimensionNumbers& dim_numbers = in HandleGather()
|
| D | hlo_instructions.h | 1364 const GatherDimensionNumbers& gather_dim_numbers,
1366 const GatherDimensionNumbers& gather_dimension_numbers() const { in gather_dimension_numbers()
1379 static GatherDimensionNumbers MakeGatherDimNumbers(
1395 std::unique_ptr<GatherDimensionNumbers> gather_dimension_numbers_;
|
| D | shape_inference.h | 284 const GatherDimensionNumbers& gather_dim_numbers,
|
| D | indexed_array_analysis.h | 265 const Shape& shape, const GatherDimensionNumbers& dim_numbers,
|
| D | hlo_instruction.h | 728 const GatherDimensionNumbers& gather_dim_numbers,
1593 const GatherDimensionNumbers& gather_dimension_numbers() const;
|
| D | triangular_solve_expander.cc | 74 GatherDimensionNumbers dim_numbers; in DiagonalBlocks()
|
| D | hlo_instructions.cc | 2311 const GatherDimensionNumbers& gather_dim_numbers, in HloGatherInstruction()
2317 absl::make_unique<GatherDimensionNumbers>(gather_dim_numbers); in HloGatherInstruction()
2340 /* static */ GatherDimensionNumbers HloGatherInstruction::MakeGatherDimNumbers( in MakeGatherDimNumbers()
2344 GatherDimensionNumbers gather_dim_numbers; in MakeGatherDimNumbers()
|
| D | hlo_cost_analysis_test.cc | 650 GatherDimensionNumbers dim_numbers; in TEST_F()
|
| D | hlo_instruction.cc | 525 std::unique_ptr<GatherDimensionNumbers> gather_dimension_numbers = in CreateFromProto()
526 absl::make_unique<GatherDimensionNumbers>( in CreateFromProto()
1308 const GatherDimensionNumbers& gather_dim_numbers, in CreateGather()
3516 const GatherDimensionNumbers& HloInstruction::gather_dimension_numbers() const { in gather_dimension_numbers()
|
| D | hlo.proto | 153 xla.GatherDimensionNumbers gather_dimension_numbers = 33;
|
| D | indexed_array_analysis.cc | 254 const Shape& shape, const GatherDimensionNumbers& dim_numbers, in ComputeArrayForGather()
|
| D | shape_inference.cc | 2868 const GatherDimensionNumbers& dim_numbers) { in ValidateGatherDimensionNumbers()
2959 const GatherDimensionNumbers& gather_dim_numbers, in InferGatherShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 154 GatherDimensionNumbers gather_dnums; in TorchGather()
171 GatherDimensionNumbers gather_dnums; in TorchIndexSelect()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 100 xla::GatherDimensionNumbers dnums; in Compile()
|
| D | gather_op.cc | 116 xla::GatherDimensionNumbers dim_numbers; in XlaGather()
|
| D | resampler_ops.cc | 140 xla::GatherDimensionNumbers gather_dim_numbers; in Gather2by2Neighbors()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_data.i | 576 %typemap(in) const GatherDimensionNumbers&
577 (GatherDimensionNumbers dimension_numbers) {
|
| D | local_computation_builder.h | 374 const GatherDimensionNumbers& dimension_numbers,
|
| D | local_computation_builder.cc | 756 const GatherDimensionNumbers& dimension_numbers, in Gather()
|
| D | xla_client.py | 1949 class GatherDimensionNumbers(object): class
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | gather_operation_test.cc | 646 GatherDimensionNumbers dim_numbers; in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 544 const GatherDimensionNumbers& dimension_numbers,
969 const GatherDimensionNumbers& dimension_numbers,
1816 const GatherDimensionNumbers& dimension_numbers,
|
| D | xla_builder.cc | 1841 const GatherDimensionNumbers& dimension_numbers, in Gather()
3425 const GatherDimensionNumbers& dimension_numbers, in Gather()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 438 message GatherDimensionNumbers { message
468 // GatherDimensionNumbers. Differences are noted below.
|