Searched refs:shape_and_slices_flat (Results 1 – 3 of 3) sorted by relevance
106 const auto& shape_and_slices_flat = shape_and_slices.flat<string>(); in Compute() local116 if (!shape_and_slices_flat(i).empty()) { in Compute()117 const string& shape_spec = shape_and_slices_flat(i); in Compute()
324 const auto& shape_and_slices_flat = shape_and_slices.flat<string>(); in RestoreTensorsV2() local362 const string& shape_and_slice = shape_and_slices_flat(i); in RestoreTensorsV2()
103 const auto& shape_and_slices_flat = in __anonec23b74b0302() local105 if (shape_and_slices_flat.size() != c->num_outputs()) { in __anonec23b74b0302()109 for (int i = 0; i < shape_and_slices_flat.size(); ++i) { in __anonec23b74b0302()110 const string& shape_and_slice = shape_and_slices_flat(i); in __anonec23b74b0302()