Searched refs:element_part (Results 1 – 1 of 1) sorted by relevance
374 xla::XlaOp element_part = xla::GetTupleElement(element, i); in ExecuteTensorListPushBack() local378 element_part = xla::Reshape(element_part, element_part_dims); in ExecuteTensorListPushBack()387 xla::DynamicUpdateSlice(list_part, element_part, start_indices); in ExecuteTensorListPushBack()