Searched refs:updates_flat (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op.cc | 132 auto updates_flat = in DoCompute() local
137 params_flat, updates_flat, indices_flat); in DoCompute()
|
| D | resource_variable_ops.cc | 924 auto updates_flat = updates.shaped<T, 2>({N, num_updates / N}); in DoCompute() local
928 params_flat, updates_flat, indices_flat); in DoCompute()
|
| D | scatter_nd_op.cc | 913 auto updates_flat = updates.shaped<T, 2>({num_updates, slice_size}); in DoScatterNd() local
951 output_matrix, indices_flat, updates_flat, output_matrix); \ in DoScatterNd()
|