| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op.cc | 124 auto indices_flat = indices.flat<Index>(); in DoCompute() local
132 params_flat, update, indices_flat); in DoCompute()
136 " = ", indices_flat(bad_i), " is not in [0, ", in DoCompute()
144 params_flat, updates_flat, indices_flat); in DoCompute()
148 " = ", indices_flat(bad_i), " is not in [0, ", in DoCompute()
212 auto indices_flat = indices_host.flat<Index>(); in DoCompute() local
220 params_flat, update, indices_flat); in DoCompute()
224 " = ", indices_flat(bad_i), " is not in [0, ", in DoCompute()
232 params_flat, updates_flat, indices_flat); in DoCompute()
236 " = ", indices_flat(bad_i), " is not in [0, ", in DoCompute()
|
| D | dynamic_stitch_op.cc | 188 GpuDeviceArrayOnHost<int32> indices_flat(c, first_dim_size); in Compute() local
190 OP_REQUIRES_OK(c, indices_flat.Init()); in Compute()
194 indices_flat.Set(i, -1); in Compute()
208 indices_flat.Set(indices_vec(j), base_size + j); in Compute()
216 OP_REQUIRES_OK(c, indices_flat.Finalize()); in Compute()
221 indices_flat.data(), data_flat.data(), output); in Compute()
|
| D | gather_op.cc | 107 auto indices_flat = indices.flat<Index>(); in Compute() local
111 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
117 indices_flat(bad_i), " is not in [0, ", gather_dim_size, ")")); in Compute()
|
| D | resource_variable_ops.cc | 600 auto indices_flat = indices->flat<Index>(); in AddBatchOffsets() local
606 indices_flat(dest_idx++) += batch_offset * batch_idx; in AddBatchOffsets()
690 const auto indices_flat = op_indices->flat<Index>(); in Compute() local
694 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
700 indices_flat(bad_i), " is not in [0, ", params.dim_size(0), ")")); in Compute()
785 auto indices_flat = indices.flat<Index>(); in Compute() local
792 params_flat, update, indices_flat); in Compute()
796 " = ", indices_flat(bad_i), " is not in [0, ", in Compute()
809 params_flat, updates_flat, indices_flat); in Compute()
813 " = ", indices_flat(bad_i), " is not in [0, ", in Compute()
|
| D | segment_reduction_ops_test.cc | 124 auto indices_flat = indices.flat<int32>(); in SparseSegmentMeanGradHelper() local
133 indices_flat(i) = (i * 31) % kUniqueIndices; in SparseSegmentMeanGradHelper()
|
| D | scatter_nd_op.cc | 672 auto indices_flat = index_flattener(c, indices); in DoScatterNd() local
706 output_matrix, indices_flat, updates_flat, output_matrix); \ in DoScatterNd()
731 gtl::ArraySlice<Index>(&indices_flat(bad_i, 0), slice_dim), ", "), in DoScatterNd()
|
| /external/tensorflow/tensorflow/examples/label_image/ |
| D | main.cc | 247 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopLabels() local
249 const int label_index = indices_flat(pos); in PrintTopLabels()
265 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in CheckTopLabel() local
266 if (indices_flat(0) != expected) { in CheckTopLabel()
268 << indices_flat(0); in CheckTopLabel()
|
| /external/tensorflow/tensorflow/contrib/pi_examples/label_image/ |
| D | label_image.cc | 272 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopLabels() local
274 const int label_index = indices_flat(pos); in PrintTopLabels()
290 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in CheckTopLabel() local
291 if (indices_flat(0) != expected) { in CheckTopLabel()
293 << indices_flat(0); in CheckTopLabel()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | segment_reduction_ops_test.py | 289 indices_flat = np.array([0, 4, 0, 8, 3, 8, 4, 7, 7, 3])
291 for indices in indices_flat, indices_flat.reshape(5, 2):
318 indices_flat = np.array([0, 4, 0, 8, 3, 8, 4, 7, 7, 3])
320 for indices in indices_flat, indices_flat.reshape(5, 2):
340 indices_flat = np.array([0, 4, 0, -1, 3, -1, 4, 7, 7, 3])
341 num_segments = max(indices_flat) + 3
344 for indices in indices_flat, indices_flat.reshape(5, 2):
446 indices_flat = np.array([0, 4, 0, 8, 3, 8, 4, 7, 7, 3])
448 for indices in indices_flat, indices_flat.reshape(5, 2):
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | label_wav.cc | 167 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in main() local
169 const int label_index = indices_flat(pos); in main()
|
| /external/tensorflow/tensorflow/examples/multibox_detector/ |
| D | main.cc | 286 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopDetections() local
299 const int label_index = indices_flat(pos); in PrintTopDetections()
|
| /external/tensorflow/tensorflow/contrib/pi_examples/camera/ |
| D | camera.cc | 316 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopLabels() local
318 const int label_index = indices_flat(pos); in PrintTopLabels()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1502 auto indices_flat = out_indices->flat<int64>(); in FastParseSingleExample() local
1504 indices_flat(i) = static_cast<int64>(i); in FastParseSingleExample()
|