Searched refs:params_flat (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op.cc | 125 auto params_flat = params.flat_outer_dims<T>(); in DoCompute() local
132 params_flat, update, indices_flat); in DoCompute()
144 params_flat, updates_flat, indices_flat); in DoCompute()
213 auto params_flat = params.flat_outer_dims<T>(); in DoCompute() local
220 params_flat, update, indices_flat); in DoCompute()
232 params_flat, updates_flat, indices_flat); in DoCompute()
|
| D | gather_op.cc | 105 auto params_flat = in Compute() local
111 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
|
| D | gather_nd_op.cc | 169 auto params_flat = params.flat_outer_dims<T, IXDIM + 1>(); \ in DoGatherNd()
171 params_flat, indices_mat, out_mat); \ in DoGatherNd()
|
| D | resource_variable_ops.cc | 689 auto params_flat = params.shaped<T, 3>({1, gather_dim_size, inner_size}); in Compute() local
694 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
786 auto params_flat = params->flat_outer_dims<T>(); in Compute() local
792 params_flat, update, indices_flat); in Compute()
809 params_flat, updates_flat, indices_flat); in Compute()
|