| /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, ")"));
|
| 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), ")"));
|
| dynamic_stitch_op.cc | 176 CudaDeviceArrayOnHost<int32> indices_flat(c, first_dim_size);
178 OP_REQUIRES_OK(c, indices_flat.Init());
180 // initialize the indices_flat (-1 represents missing indices)
182 indices_flat.Set(i, -1);
193 // indices_flat's indices represent the indices of output.
194 // indices_flat's values represent the indices of input_data where the
196 indices_flat.Set(indices_vec(j), base_size + j);
204 OP_REQUIRES_OK(c, indices_flat.Finalize());
209 indices_flat.data(), data_flat.data(), output);
|
| segment_reduction_ops_test.cc | 124 auto indices_flat = indices.flat<int32>(); local
133 indices_flat(i) = (i * 31) % kUniqueIndices;
|
| 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/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/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/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/python/kernel_tests/ |
| segment_reduction_ops_test.py | 275 indices_flat = np.array([0, 4, 0, 8, 3, 8, 4, 7, 7, 3])
277 for indices in indices_flat, indices_flat.reshape(5, 2):
304 indices_flat = np.array([0, 4, 0, 8, 3, 8, 4, 7, 7, 3])
306 for indices in indices_flat, indices_flat.reshape(5, 2):
325 indices_flat = np.array([0, 4, 0, -1, 3, -1, 4, 7, 7, 3])
326 num_segments = max(indices_flat) + 3
329 for indices in indices_flat, indices_flat.reshape(5, 2)
[all...] |
| /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/core/util/ |
| example_proto_fast_parsing.cc | 1385 auto indices_flat = out_indices->flat<int64>(); local
[all...] |
