Searched refs:shape_and_slice_specs (Results 1 – 1 of 1) sorted by relevance
138 std::vector<std::vector<std::string>> shape_and_slice_specs; in BuildPerDeviceSave() local150 shape_and_slice_specs.push_back({}); in BuildPerDeviceSave()157 shape_and_slice_specs[save_op_index].push_back(specs[save_op_index]); in BuildPerDeviceSave()161 return SaveOpSpecs(new_prefixes, tensor_indices, shape_and_slice_specs); in BuildPerDeviceSave()