| /external/tensorflow/tensorflow/core/kernels/ |
| gather_op.cc | 107 auto indices_flat = indices.flat<Index>(); variable
111 int64 bad_i = functor(c, params_flat, indices_flat, out_flat);
117 indices_flat(bad_i), " is not in [0, ", gather_dim_size, ")"));
|
| segment_reduction_ops_test.cc | 124 auto indices_flat = indices.flat<int32>(); local
133 indices_flat(i) = (i * 31) % kUniqueIndices;
|
| scatter_op.cc | 123 auto indices_flat = indices.flat<Index>(); local
129 params_flat, updates_flat, indices_flat);
134 indices_flat(bad_i), " is not in [0, ", params.dim_size(0), ")"));
196 auto indices_flat = indices_host.flat<Index>(); local
202 params_flat, updates_flat, indices_flat);
207 indices_flat(bad_i), " is not in [0, ", params.dim_size(0), ")"));
|
| resource_variable_ops.cc | 520 auto indices_flat = indices.flat<Index>(); variable
524 int64 bad_i = functor(c, params_flat, indices_flat, out_flat);
530 indices_flat(bad_i), " is not in [0, ", params.dim_size(0), ")"));
598 auto indices_flat = indices.flat<Index>(); variable
604 params_flat, updates_flat, indices_flat);
608 " = ", indices_flat(bad_i), " is not in [0, ",
|
| scatter_nd_op.cc | 447 auto indices_flat = index_flattener(c, indices); local
481 output_matrix, indices_flat, updates_flat, output_matrix); \
504 gtl::ArraySlice<Index>(&indices_flat(bad_i, 0), slice_dim), ", "),
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| label_wav.cc | 167 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); local
169 const int label_index = indices_flat(pos);
|
| /external/tensorflow/tensorflow/examples/label_image/ |
| main.cc | 246 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); local
248 const int label_index = indices_flat(pos);
264 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); local
265 if (indices_flat(0) != expected) {
267 << indices_flat(0);
|
| /external/tensorflow/tensorflow/examples/multibox_detector/ |
| main.cc | 282 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); local
295 const int label_index = indices_flat(pos);
|
| /external/tensorflow/tensorflow/contrib/pi_examples/camera/ |
| camera.cc | 316 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); local
318 const int label_index = indices_flat(pos);
|
| /external/tensorflow/tensorflow/contrib/pi_examples/label_image/ |
| label_image.cc | 272 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); local
274 const int label_index = indices_flat(pos);
290 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); local
291 if (indices_flat(0) != expected) {
293 << indices_flat(0);
|
| /external/tensorflow/tensorflow/core/util/ |
| example_proto_fast_parsing.cc | 1385 auto indices_flat = out_indices->flat<int64>(); local
[all...] |
