| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 153 const GatherDimensionNumbers& dim_numbers = gather.gather_dimension_numbers(); in GatherLoopBody()
277 gather_instr->gather_dimension_numbers(); in GatherLoopTripCount()
334 gather_instr->gather_dimension_numbers(); in ExpandInstruction()
354 gather_instr->gather_dimension_numbers()); in ExpandInstruction()
|
| 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()
949 const auto& dnums = hlo.gather_dimension_numbers(); in GatherOutputShardingFromDataOperand()
963 const auto& dnums = hlo.gather_dimension_numbers(); in GatherDataOperandShardingFromOutput()
1259 const auto& dnums = hlo.gather_dimension_numbers(); in GetGatherBatchParallelDims()
1342 auto dnums = gather.gather_dimension_numbers(); in GatherParallelOutputDims()
1361 auto dnums = gather.gather_dimension_numbers(); in GatherOutputAlignedOperandParallelDims()
|
| D | hlo_instructions.cc | 2737 const GatherDimensionNumbers& gather_dimension_numbers) { in GatherDimensionNumbersToString() argument
2740 StrJoin(gather_dimension_numbers.offset_dims(), ","), "}"); in GatherDimensionNumbersToString()
2743 StrJoin(gather_dimension_numbers.collapsed_slice_dims(), ","), "}"); in GatherDimensionNumbersToString()
2746 StrJoin(gather_dimension_numbers.start_index_map(), ","), "}"); in GatherDimensionNumbersToString()
2748 StrCat("index_vector_dim=", gather_dimension_numbers.index_vector_dim()); in GatherDimensionNumbersToString()
2776 *proto.mutable_gather_dimension_numbers() = gather_dimension_numbers(); in ToProto()
2787 GatherDimensionNumbersToString(gather_dimension_numbers()), in ExtraAttributesToStringImpl()
2801 gather_dimension_numbers(), in IdenticalSlowPath()
2802 casted_other.gather_dimension_numbers()) && in IdenticalSlowPath()
2812 shape, new_operands[0], new_operands[1], gather_dimension_numbers(), in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.cc | 668 auto gather_dimension_numbers = absl::make_unique<GatherDimensionNumbers>( in CreateFromProto() local
669 proto.gather_dimension_numbers()); in CreateFromProto()
675 *gather_dimension_numbers, gather_slice_sizes, in CreateFromProto()
4236 const GatherDimensionNumbers& HloInstruction::gather_dimension_numbers() const { in gather_dimension_numbers() function in xla::HloInstruction
4237 return Cast<HloGatherInstruction>(this)->gather_dimension_numbers(); in gather_dimension_numbers()
|
| D | hlo_evaluator.cc | 1721 gather->gather_dimension_numbers(); in HandleGather()
1747 &gather->gather_dimension_numbers(), /*input_shape=*/operand.shape(), in HandleGather()
1750 gather->gather_dimension_numbers(), /*input_shape=*/operand.shape(), in HandleGather()
|
| D | hlo_instructions.h | 1659 const GatherDimensionNumbers& gather_dimension_numbers() const { in gather_dimension_numbers() function
1680 const GatherDimensionNumbers& gather_dimension_numbers);
|
| D | indexed_array_analysis.cc | 155 ComputeArrayForGather(instr->shape(), instr->gather_dimension_numbers(), in ComputeArrayFor()
|
| D | hlo.proto | 157 xla.GatherDimensionNumbers gather_dimension_numbers = 33; field
|
| D | hlo_instruction.h | 1883 const GatherDimensionNumbers& gather_dimension_numbers() const;
|
| D | dynamic_dimension_inference.cc | 1194 hlo->gather_dimension_numbers(); in HandleGather()
|
| D | sharding_propagation.cc | 414 auto dnums = instruction->gather_dimension_numbers(); in InferGatherParallelShardingFromOperands()
|
| D | hlo_verifier.cc | 1136 gather->gather_dimension_numbers(), gather->gather_slice_sizes())); in HandleGather()
|
| D | algebraic_simplifier.cc | 2533 gather->gather_dimension_numbers().index_vector_dim() == in HandleGather()
2535 gather->gather_dimension_numbers().collapsed_slice_dims_size() == 1) { in HandleGather()
|
| D | elemental_ir_emitter.cc | 1914 const GatherDimensionNumbers& dim_numbers = hlo->gather_dimension_numbers(); in EmitElementalGather()
|
| /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()
673 const auto& dnums = gather->gather_dimension_numbers(); in HandleGather()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 584 ? use->gather_dimension_numbers().start_index_map() in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 351 gather_instruction->gather_dimension_numbers(), builder_))); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1246 hlo_gather->gather_dimension_numbers(), &builder_)); in EmitGatherOp()
|