| /external/tensorflow/tensorflow/core/kernels/ |
| gather_functor_gpu.cu.h | 34 int64 gather_dim_size, int64 indices_size,
46 batch_i = batch_indices_i / indices_size;
48 indices_i = batch_indices_i - batch_i * indices_size;
90 const int64 indices_size = indices.size(); local
98 out.data(), gather_dim_size, indices_size, slice_size, out_size));
103 out.data(), gather_dim_size, indices_size, slice_size, out_size));
|
| scatter_functor_gpu.cu.h | 76 Index indices_size) {
77 Index update_block = updates_size / indices_size;
96 Index indices_size,
98 Index update_block = synthesized_updates_size / indices_size;
127 const Index indices_size = indices.size(); local
133 indices.data(), first_dim_size, updates_size, indices_size));
148 const Index indices_size = indices.size(); local
149 const Index synthesized_updates_size = indices_size * params.dimension(1);
155 indices_size, synthesized_updates_size));
|
| gather_functor.h | 44 const SliceIndex indices_size = static_cast<SliceIndex>(indices.dimension(0)); local
61 SliceIndex batch_idx = static_cast<SliceIndex>(start / indices_size);
62 SliceIndex indices_idx = static_cast<SliceIndex>(start % indices_size);
63 SliceIndex batch_idx_end = static_cast<SliceIndex>(end / indices_size);
64 SliceIndex indices_idx_end = static_cast<SliceIndex>(end % indices_size);
71 (i_next < indices_size)) {
93 out_base + (batch_idx * indices_size + indices_idx) * slice_elems,
108 batch_size * indices_size, slice_elems * sizeof(T), work);
|
| gather_nd_op_gpu.cu.cc | 33 const Eigen::array<int64, IXDIM> batch_indices, const int64 indices_size,
74 const int64 indices_size = Tindices.dimension(1); local
93 indices_size, s_size, out_size));
|
| dynamic_partition_op_gpu.cu.cc | 104 T* out, int64 gather_dim_size, int64 indices_size,
110 gather_dim_size, indices_size, slice_size, out_size));
446 int32 indices_size = outs[p]->dim_size(0); local
451 indices_size, slice_size, out_size);
452 ind_base += indices_size;
|
| /external/tensorflow/tensorflow/python/ops/ |
| array_grad.py | 429 indices_size = array_ops.expand_dims(array_ops.size(indices), 0)
439 values_shape = array_ops.concat([indices_size, params_tail_shape], 0)
441 indices = array_ops.reshape(indices, indices_size)
453 values_shape = array_ops.concat([outer_shape, indices_size, inner_shape], 0)
455 indices = array_ops.reshape(indices, indices_size)
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| gather_op_test.py | 416 indices_size = np.prod(indices_shape)
417 indices = np.reshape(np.arange(indices_size), indices_shape)
|
| resource_variable_ops_test.py | [all...] |
