Home
last modified time | relevance | path

Searched refs:shape_and_slices_flat (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dio_ops.cc108 const auto& shape_and_slices_flat = in __anon711ce7c10302() local
110 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()
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_v2_ops.cc109 const auto& shape_and_slices_flat = shape_and_slices.flat<tstring>(); in Compute() local
120 if (!shape_and_slices_flat(i).empty()) { in Compute()
121 const string& shape_spec = shape_and_slices_flat(i); in Compute()
Dsave_restore_tensor.cc371 const auto& shape_and_slices_flat = shape_and_slices.flat<tstring>(); in RestoreTensorsV2() local
377 shape_and_slices_flat(i), prefix_string, dtypes[i]}); in RestoreTensorsV2()