Searched refs:element_part_dims (Results 1 – 1 of 1) sorted by relevance
375 std::vector<int64> element_part_dims = in ExecuteTensorListPushBack() local377 element_part_dims.insert(element_part_dims.begin(), 1); in ExecuteTensorListPushBack()378 element_part = xla::Reshape(element_part, element_part_dims); in ExecuteTensorListPushBack()