Searched refs:gather_slice_sizes (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 202 gather.gather_slice_sizes())); in GatherLoopBody()
344 gather_instr->gather_slice_sizes(), gather_loop_trip_count, in ExpandInstruction()
|
| D | hlo_instructions.cc | 2362 for (int64 bound : gather_slice_sizes()) { in ToProto()
2371 StrCat("slice_sizes={", StrJoin(gather_slice_sizes(), ","), "}")}; in ExtraAttributesToStringImpl()
2382 gather_slice_sizes() == casted_other.gather_slice_sizes(); in IdenticalSlowPath()
2391 gather_slice_sizes()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.cc | 528 std::vector<int64> gather_slice_sizes; in CreateFromProto() local
529 for (int64 bound : proto.gather_slice_sizes()) { in CreateFromProto()
530 gather_slice_sizes.push_back(bound); in CreateFromProto()
533 *gather_dimension_numbers, gather_slice_sizes); in CreateFromProto()
3520 absl::Span<const int64> HloInstruction::gather_slice_sizes() const { in gather_slice_sizes() function in xla::HloInstruction
3521 return Cast<HloGatherInstruction>(this)->gather_slice_sizes(); in gather_slice_sizes()
|
| D | hlo.proto | 154 repeated int64 gather_slice_sizes = 34; field
|
| D | indexed_array_analysis.cc | 156 instr->gather_slice_sizes(), in ComputeArrayFor()
|
| D | hlo_instruction.h | 1595 absl::Span<const int64> gather_slice_sizes() const;
|
| D | hlo_verifier.cc | 815 gather->gather_dimension_numbers(), gather->gather_slice_sizes())); in HandleGather()
|
| D | hlo_instructions.h | 1370 absl::Span<const int64> gather_slice_sizes() const { in gather_slice_sizes() function
|
| D | hlo_evaluator.cc | 1068 shape.dimensions_size(), gather->gather_slice_sizes(), dim_numbers); in HandleGather()
|