Searched refs:updates_flat (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op.cc | 139 auto updates_flat = in DoCompute() local
144 params_flat, updates_flat, indices_flat); in DoCompute()
227 auto updates_flat = in DoCompute() local
232 params_flat, updates_flat, indices_flat); in DoCompute()
|
| D | scatter_nd_op.cc | 673 auto updates_flat = updates.shaped<T, 2>({num_updates, slice_size}); in DoScatterNd() local
706 output_matrix, indices_flat, updates_flat, output_matrix); \ in DoScatterNd()
|
| D | resource_variable_ops.cc | 805 auto updates_flat = updates.shaped<T, 2>({N, num_updates / N}); in Compute() local
809 params_flat, updates_flat, indices_flat); in Compute()
|