Searched refs:dims_list (Results 1 – 2 of 2) sorted by relevance
59 std::vector<std::vector<int64_t>> dims_list) { in AllPermutationsOfShapes() argument61 for (const auto& dims : dims_list) { in AllPermutationsOfShapes()
296 for dims_list in self._GetParamsCached().input_dims:297 assert dims_list, f"Expect non-empty `dim_list` but got: {dims_list}"299 input_batches = [dims[0] for dims in dims_list]