HomeSort by relevance Sort by last modified time
    Searched refs:indices_size (Results 1 - 6 of 6) sorted by null

  /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
99 indices_size, slice_size, out_size);
106 indices_size, slice_size, out_size);
scatter_functor_gpu.cu.h 36 Index indices_size) {
37 Index update_block = updates_size / indices_size;
84 const Index indices_size = indices.size(); local
90 updates_size, indices_size);
gather_functor.h 42 const SliceIndex indices_size = static_cast<SliceIndex>(indices.dimension(0)); local
59 SliceIndex batch_idx = static_cast<SliceIndex>(start / indices_size);
60 SliceIndex indices_idx = static_cast<SliceIndex>(start % indices_size);
61 SliceIndex batch_idx_end = static_cast<SliceIndex>(end / indices_size);
62 SliceIndex indices_idx_end = static_cast<SliceIndex>(end % indices_size);
69 (i_next < indices_size)) {
91 out_base + (batch_idx * indices_size + indices_idx) * slice_elems,
106 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 batch_indices, indices_size, s_size, out_size);
dynamic_partition_op_gpu.cu.cc 103 T* out, int64 gather_dim_size, int64 indices_size,
108 params, indices, out, gather_dim_size, indices_size, slice_size,
445 int32 indices_size = outs[p]->dim_size(0); local
450 indices_size, slice_size, out_size);
451 ind_base += indices_size;
  /external/tensorflow/tensorflow/python/ops/
array_grad.py 427 indices_size = array_ops.expand_dims(array_ops.size(indices), 0)
437 values_shape = array_ops.concat([indices_size, params_tail_shape], 0)
439 indices = array_ops.reshape(indices, indices_size)
451 values_shape = array_ops.concat([outer_shape, indices_size, inner_shape], 0)
453 indices = array_ops.reshape(indices, indices_size)

Completed in 86 milliseconds