Searched refs:params_flat (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_op.cc | 165 auto params_flat = params.shaped<T, 4>( in Compute() local
171 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
173 auto params_flat = in Compute() local
178 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
|
| D | scatter_op.cc | 119 auto params_flat = params.flat_outer_dims<T>(); in DoCompute() local
125 params_flat, update, indices_flat); in DoCompute()
137 params_flat, updates_flat, indices_flat); in DoCompute()
|
| D | gather_nd_op.h | 136 auto params_flat = params.flat_outer_dims<T, IXDIM + 1>(); \ in DoGatherNd()
138 params_flat, indices_mat, out_mat); \ in DoGatherNd()
|
| D | resource_variable_ops.cc | 693 auto params_flat = params.shaped<T, 3>({1, gather_dim_size, inner_size}); in Compute() local
698 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute()
905 auto params_flat = params->flat_outer_dims<T>(); in DoCompute() local
911 params_flat, update, indices_flat); in DoCompute()
928 params_flat, updates_flat, indices_flat); in DoCompute()
|