Home
last modified time | relevance | path

Searched refs:new_tuple_index (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment.cc3726 int new_tuple_index = calling_instruction_->shape().tuple_shapes_size(); in Process() local
3732 {new_tuple_index})); in Process()
3742 defining_position_.index = {new_tuple_index}; in Process()
3748 TupleUtil::ExtractPrefix(calling_instruction_, new_tuple_index); in Process()
H A Dhlo_instructions.h1453 void set_tuple_index(int64_t new_tuple_index) { in set_tuple_index() argument
1454 tuple_index_ = new_tuple_index; in set_tuple_index()
H A Dhlo_instruction.h2002 void set_tuple_index(int64_t new_tuple_index);
H A Dhlo_instruction.cc4663 void HloInstruction::set_tuple_index(int64_t new_tuple_index) { in set_tuple_index() argument
4665 new_tuple_index); in set_tuple_index()