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

  /external/tensorflow/tensorflow/core/kernels/
gather_nd_op_gpu.cu.cc 32 const Eigen::array<int64, IXDIM> batch_strides,
46 offset += batch_strides[j] * index_j;
77 Eigen::array<int64, IXDIM> batch_strides; local
80 batch_strides[size_t(IXDIM - 1)] = s_size;
85 batch_strides[i - 1] = batch_strides[i] * Tparams.dimension(i);
92 Tparams.data(), Tindices.data(), Tout.data(), batch_strides,
scatter_nd_op_cpu_impl.h 105 Index batch_strides[IXDIM]; local
108 batch_strides[dim] = 1;
110 batch_strides[dim] =
111 batch_strides[dim + 1] * output_shape_prefix[dim + 1];
121 i += ix_d * batch_strides[dim];
187 Index batch_strides[IXDIM]; local
190 batch_strides[dim] = 1;
192 batch_strides[dim] =
193 batch_strides[dim + 1] * output_shape_prefix[dim + 1];
203 i += ix_d * batch_strides[dim]
    [all...]
scatter_nd_op_gpu.cu.cc 85 const Eigen::array<int64, IXDIM> batch_strides, const int64 num_indices,
97 i += ix_d * batch_strides[dim] * slice_size;
127 // Index batch_strides[IXDIM];
128 Eigen::array<int64, IXDIM> batch_strides; local
131 batch_strides[dim] = 1;
133 batch_strides[dim] =
134 batch_strides[dim + 1] * output_shape_prefix[dim + 1];
143 batch_strides, batch_size, slice_size);

Completed in 365 milliseconds