Searched refs:retval_index_and_layout (Results 1 – 1 of 1) sorted by relevance
191 std::vector<std::pair<int64, xla::Layout>> retval_index_and_layout; in BuildComputation() local220 retval_index_and_layout.emplace_back(elems.size(), shape.layout()); in BuildComputation()295 retval_index_and_layout.emplace_back( in BuildComputation()327 for (auto& index_and_layout : retval_index_and_layout) { in BuildComputation()