Searched refs:GatherDimensionNumbers (Results 1 – 25 of 34) sorted by relevance
12
| /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_instructions.h | 1451 const GatherDimensionNumbers& gather_dim_numbers,
1453 const GatherDimensionNumbers& gather_dimension_numbers() const { in gather_dimension_numbers()
1468 static GatherDimensionNumbers MakeGatherDimNumbers(
1474 const GatherDimensionNumbers& gather_dimension_numbers);
1487 std::unique_ptr<GatherDimensionNumbers> gather_dimension_numbers_;
|
| D | indexed_array_analysis.h | 265 const Shape& shape, const GatherDimensionNumbers& dim_numbers,
|
| D | shape_inference.h | 289 const GatherDimensionNumbers& gather_dim_numbers,
|
| D | hlo_evaluator.cc | 1439 const Shape& output_shape, const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputBatchIndices()
1458 const GatherDimensionNumbers& dim_numbers) { in IterationSpaceForOutputOffsetIndices()
1488 const GatherDimensionNumbers* dim_numbers, const Shape& input_shape, in OutputBatchIndexToInputIndex()
1604 const GatherDimensionNumbers& dim_numbers_;
1616 const GatherDimensionNumbers& dim_numbers, const Shape& input_shape, in OutputOffsetIndexToInputIndex()
1708 const GatherDimensionNumbers& dim_numbers = in HandleGather()
|
| D | hlo_instructions.cc | 2441 const GatherDimensionNumbers& gather_dim_numbers, in HloGatherInstruction()
2448 absl::make_unique<GatherDimensionNumbers>(gather_dim_numbers); in HloGatherInstruction()
2453 const GatherDimensionNumbers& gather_dimension_numbers) { in GatherDimensionNumbersToString()
2471 /* static */ GatherDimensionNumbers HloGatherInstruction::MakeGatherDimNumbers( in MakeGatherDimNumbers()
2475 GatherDimensionNumbers gather_dim_numbers; in MakeGatherDimNumbers()
|
| D | hlo_instruction.h | 875 const GatherDimensionNumbers& gather_dim_numbers,
1771 const GatherDimensionNumbers& gather_dimension_numbers() const;
|
| D | triangular_solve_expander.cc | 74 GatherDimensionNumbers dim_numbers; in DiagonalBlocks()
|
| D | hlo.proto | 157 xla.GatherDimensionNumbers gather_dimension_numbers = 33;
|
| D | indexed_array_analysis.cc | 254 const Shape& shape, const GatherDimensionNumbers& dim_numbers, in ComputeArrayForGather()
|
| D | hlo_cost_analysis_test.cc | 870 GatherDimensionNumbers dim_numbers; in 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/mlir/xla/ |
| D | hlo_function_importer.h | 122 const xla::GatherDimensionNumbers& dnums);
|
| D | mlir_hlo_to_hlo.cc | 324 static xla::GatherDimensionNumbers Convert_dimension_numbers( in Convert_dimension_numbers()
325 mlir::xla_hlo::GatherDimensionNumbers input) { in Convert_dimension_numbers()
326 xla::GatherDimensionNumbers output; in Convert_dimension_numbers()
|
| D | hlo_function_importer.cc | 793 const xla::GatherDimensionNumbers& dnums) { in ConvertGatherDimensionNumbers()
800 auto attr = mlir::xla_hlo::GatherDimensionNumbers::get( in ConvertGatherDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 201 GatherDimensionNumbers gather_dnums; in TorchGather()
264 GatherDimensionNumbers gather_dnums; in TorchIndexSelect()
|
| D | matrix.cc | 148 xla::GatherDimensionNumbers dim_numbers; in GetMatrixDiagonalViaGather()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | types.h | 334 struct type_caster<xla::GatherDimensionNumbers> {
336 PYBIND11_TYPE_CASTER(xla::GatherDimensionNumbers,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | gather_operation_test.cc | 645 GatherDimensionNumbers dim_numbers; in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | hlo_ops.td | 955 def GatherDimensionNumbers : StructAttr<"GatherDimensionNumbers", HLO_Dialect,
967 GatherDimensionNumbers:$dimension_numbers,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 583 const GatherDimensionNumbers& dimension_numbers,
1005 const GatherDimensionNumbers& dimension_numbers,
1883 const GatherDimensionNumbers& dimension_numbers,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 451 message GatherDimensionNumbers { message
481 // GatherDimensionNumbers. Differences are noted below.
|
12