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

  /external/tensorflow/tensorflow/core/kernels/
gather_functor_gpu.cu.h 34 int64 gather_dim_size, int64 indices_size,
56 // Check gather_i is in [0, gather_dim_size).
57 if (!FastBoundsCheck(gather_i, gather_dim_size)) {
62 // params is a [batch_size, gather_dim_size, slice_size] tensor. Read
66 (batch_i * gather_dim_size + gather_i) * slice_size + slice_i;
89 const int64 gather_dim_size = params.dimension(1); local
98 params.data(), indices.data(), out.data(), gather_dim_size,
105 params.data(), indices.data(), out.data(), gather_dim_size,
gather_op.cc 78 const int64 gather_dim_size = params.dim_size(axis); variable
81 c, gather_dim_size <= std::numeric_limits<Index>::max(),
84 " indexing: ", gather_dim_size, " > ",
106 params.shaped<T, 3>({outer_size, gather_dim_size, inner_size});
117 indices_flat(bad_i), " is not in [0, ", gather_dim_size, ")"));
resource_variable_ops.cc 514 const int64 gather_dim_size = params.dim_size(0); variable
519 auto params_flat = params.shaped<T, 3>({1, gather_dim_size, inner_size});

Completed in 87 milliseconds