Searched refs:list_tuple_size (Results 1 – 1 of 1) sorted by relevance
363 int list_tuple_size = xla::ShapeUtil::TupleElementCount(list_shape); in ExecuteTensorListPushBack() local364 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPushBack()428 int list_tuple_size = xla::ShapeUtil::TupleElementCount(list_shape); in ExecuteTensorListPopBack() local429 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPopBack()433 for (int i = 0; i < list_tuple_size - 1; i++) { in ExecuteTensorListPopBack()