| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 92 auto indices_flat = indices->template flat<int64>(); in Compute() local
115 indices_flat(indices_offset) = row_idx; in Compute()
116 indices_flat(indices_offset + 1) = col_idx; in Compute()
118 indices_flat(indices_offset) = batch_idx; in Compute()
119 indices_flat(indices_offset + 1) = row_idx; in Compute()
120 indices_flat(indices_offset + 2) = col_idx; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op.cc | 118 auto indices_flat = indices.flat<Index>(); in DoCompute() local
125 params_flat, update, indices_flat); in DoCompute()
129 " = ", indices_flat(bad_i), " is not in [0, ", in DoCompute()
137 params_flat, updates_flat, indices_flat); in DoCompute()
141 " = ", 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 | 163 auto indices_flat = indices.flat<Index>(); in Compute() local
171 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
178 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
184 indices_flat(bad_i), " is not in [0, ", gather_dim_size, ")")); in Compute()
|
| D | resource_variable_ops.cc | 694 const auto indices_flat = op_indices->flat<Index>(); in Compute() local
698 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
704 indices_flat(bad_i), " is not in [0, ", params.dim_size(0), ")")); in Compute()
719 auto indices_flat = indices->flat<Index>(); in AddBatchOffsets() local
725 indices_flat(dest_idx++) += batch_offset * batch_idx; in AddBatchOffsets()
904 auto indices_flat = indices.flat<Index>(); in DoCompute() local
911 params_flat, update, indices_flat); in DoCompute()
915 " = ", indices_flat(bad_i), " is not in [0, ", in DoCompute()
928 params_flat, updates_flat, indices_flat); in DoCompute()
932 " = ", indices_flat(bad_i), " is not in [0, ", in DoCompute()
|
| 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 | 912 auto indices_flat = index_flattener(c, indices); in DoScatterNd() local
951 output_matrix, indices_flat, updates_flat, output_matrix); \ in DoScatterNd()
976 gtl::ArraySlice<Index>(&indices_flat(bad_i, 0), slice_dim), ", "), in DoScatterNd()
|
| D | training_ops.cc | 3429 auto indices_flat = indices.flat<Tindex>(); in Compute() local
3432 lr.scalar<T>(), grad.flat_outer_dims<T>(), indices_flat, in Compute()
3438 indices_flat(bad_i), " is not in [0, ", var.dim_size(0), ")")); in Compute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | segment_reduction_ops_test.py | 302 indices_flat = np.array([0, 4, 0, 8, 3, 8, 4, 7, 7, 3])
304 for indices in indices_flat, indices_flat.reshape(5, 2):
331 indices_flat = np.array([0, 4, 0, 8, 3, 8, 4, 7, 7, 3])
333 for indices in indices_flat, indices_flat.reshape(5, 2):
353 indices_flat = np.array([0, 4, 0, -1, 3, -1, 4, 7, 7, 3])
354 num_segments = max(indices_flat) + 3
357 for indices in indices_flat, indices_flat.reshape(5, 2):
377 indices_flat = np.array([0, 4, 0, -1, 3, -1, 4, 7, 7, 3])
378 num_segments = max(indices_flat) + 3
381 for indices in indices_flat, indices_flat.reshape(5, 2):
[all …]
|
| /external/tensorflow/tensorflow/examples/label_image/ |
| D | main.cc | 248 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopLabels() local
250 const int label_index = indices_flat(pos); in PrintTopLabels()
266 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in CheckTopLabel() local
267 if (indices_flat(0) != expected) { in CheckTopLabel()
269 << indices_flat(0); in CheckTopLabel()
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | label_wav.cc | 168 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in main() local
170 const int label_index = indices_flat(pos); in main()
|
| /external/tensorflow/tensorflow/examples/multibox_detector/ |
| D | main.cc | 290 tensorflow::TTypes<int32>::Flat indices_flat = indices.flat<int32>(); in PrintTopDetections() local
303 const int label_index = indices_flat(pos); in PrintTopDetections()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1727 auto indices_flat = out_indices->flat<int64>(); in FastParseSingleExample() local
1729 indices_flat(i) = static_cast<int64>(i); in FastParseSingleExample()
|