Searched refs:tensor_names_flat (Results 1 – 2 of 2) sorted by relevance
88 auto tensor_names_flat = tensor_names_t.flat<tstring>(); in SaveTensors() local93 std::vector<size_t> sorted_name_idx(tensor_names_flat.size()); in SaveTensors()96 [&tensor_names_flat](size_t a, size_t b) { in SaveTensors()97 return tensor_names_flat(a) < tensor_names_flat(b); in SaveTensors()101 const string& name = tensor_names_flat(i); in SaveTensors()342 const auto& tensor_names_flat = tensor_names.flat<tstring>(); in RestoreTensorsV2() local346 std::vector<size_t> sorted_name_idx(tensor_names_flat.size()); in RestoreTensorsV2()349 [&tensor_names_flat](size_t a, size_t b) { in RestoreTensorsV2()350 return tensor_names_flat(a) < tensor_names_flat(b); in RestoreTensorsV2()363 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2()[all …]
105 const auto& tensor_names_flat = tensor_names.flat<tstring>(); in Compute() local113 const string& tensor_name = tensor_names_flat(i); in Compute()