Searched refs:shape_and_slices_flat (Results 1 – 3 of 3) sorted by relevance
108 const auto& shape_and_slices_flat = in __anon711ce7c10302() local110 if (shape_and_slices_flat.size() != c->num_outputs()) { in __anon711ce7c10302()114 for (int i = 0; i < shape_and_slices_flat.size(); ++i) { in __anon711ce7c10302()115 const string& shape_and_slice = shape_and_slices_flat(i); in __anon711ce7c10302()
109 const auto& shape_and_slices_flat = shape_and_slices.flat<tstring>(); in Compute() local120 if (!shape_and_slices_flat(i).empty()) { in Compute()121 const string& shape_spec = shape_and_slices_flat(i); in Compute()
371 const auto& shape_and_slices_flat = shape_and_slices.flat<tstring>(); in RestoreTensorsV2() local377 shape_and_slices_flat(i), prefix_string, dtypes[i]}); in RestoreTensorsV2()