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