OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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