| /external/tensorflow/tensorflow/core/kernels/ |
| gather_op.cc | 105 auto params_flat = variable
111 int64 bad_i = functor(c, params_flat, indices_flat, out_flat);
|
| scatter_op.cc | 124 auto params_flat = params.flat_outer_dims<T>(); local
129 params_flat, updates_flat, indices_flat);
197 auto params_flat = params.flat_outer_dims<T>(); local
202 params_flat, updates_flat, indices_flat);
|
| resource_variable_ops.cc | 519 auto params_flat = params.shaped<T, 3>({1, gather_dim_size, inner_size}); variable
524 int64 bad_i = functor(c, params_flat, indices_flat, out_flat);
599 auto params_flat = params->flat_outer_dims<T>(); variable
604 params_flat, updates_flat, indices_flat);
|
