Searched refs:gather_dimension_numbers (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 153 const GatherDimensionNumbers& dim_numbers = gather.gather_dimension_numbers(); in GatherLoopBody()
319 gather_instr->gather_dimension_numbers(); in ExpandInstruction()
345 gather_instr->gather_dimension_numbers()); in ExpandInstruction()
|
| D | hlo_evaluator.cc | 1051 gather->gather_dimension_numbers(); in HandleGather()
1077 &gather->gather_dimension_numbers(), /*input_shape=*/operand.shape(), in HandleGather()
1080 gather->gather_dimension_numbers(), /*input_shape=*/operand.shape(), in HandleGather()
|
| D | hlo_instructions.cc | 2361 *proto.mutable_gather_dimension_numbers() = gather_dimension_numbers(); in ToProto()
2380 gather_dimension_numbers(), in IdenticalSlowPath()
2381 casted_other.gather_dimension_numbers()) && in IdenticalSlowPath()
2390 shape, new_operands[0], new_operands[1], gather_dimension_numbers(), in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.cc | 525 std::unique_ptr<GatherDimensionNumbers> gather_dimension_numbers = in CreateFromProto() local
527 proto.gather_dimension_numbers()); in CreateFromProto()
533 *gather_dimension_numbers, gather_slice_sizes); in CreateFromProto()
3516 const GatherDimensionNumbers& HloInstruction::gather_dimension_numbers() const { in gather_dimension_numbers() function in xla::HloInstruction
3517 return Cast<HloGatherInstruction>(this)->gather_dimension_numbers(); in gather_dimension_numbers()
|
| D | hlo.proto | 153 xla.GatherDimensionNumbers gather_dimension_numbers = 33; field
|
| D | indexed_array_analysis.cc | 155 ComputeArrayForGather(instr->shape(), instr->gather_dimension_numbers(), in ComputeArrayFor()
|
| D | hlo_instruction.h | 1593 const GatherDimensionNumbers& gather_dimension_numbers() const;
|
| D | hlo_verifier.cc | 815 gather->gather_dimension_numbers(), gather->gather_slice_sizes())); in HandleGather()
|
| D | hlo_instructions.h | 1366 const GatherDimensionNumbers& gather_dimension_numbers() const { in gather_dimension_numbers() function
|
| D | elemental_ir_emitter.cc | 1867 const GatherDimensionNumbers& dim_numbers = hlo->gather_dimension_numbers(); in EmitElementalGather()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 473 ? use->gather_dimension_numbers().start_index_map() in CreateLiteralForConstrainedUses()
|