OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:slice_dim
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
slice_op.cc
291
// criteria matches for
slice_dim
: if indices for slice are 0 in all dims
292
// except
slice_dim
and if sizes of all the dimensions of the slice are same
293
// as the sizes of all the dimensions of the input except
slice_dim
, then
298
int
slice_dim
) {
300
if (dim !=
slice_dim
&&
317
// Returns True if Slicing over a single dimension, and sets
slice_dim
322
int*
slice_dim
) {
325
*
slice_dim
= dim;
335
int
slice_dim
= -1;
local
342
DoesSliceShapeDifferInOnly1D(in_shape, begin, size, &
slice_dim
)) {
[
all
...]
scatter_nd_op.cc
293
// params_shape[
slice_dim
:]
296
const int64
slice_dim
=
local
303
"params_shape[
slice_dim
:], got updates.shape: ",
307
",
slice_dim
: ",
slice_dim
, ", and batch_dim: ", batch_dim);
311
if (params_shape.dims() <
slice_dim
+ (updates.dims() - batch_dim)) {
314
if (updates.dims() != batch_dim + params_shape.dims() -
slice_dim
) {
322
params_shape.dim_size(d +
slice_dim
)) {
332
int64*
slice_dim
, Index* num_updates,
373
*
slice_dim
= (indices_shape.dims() > 1
440
int64
slice_dim
;
local
[
all
...]
/external/tensorflow/tensorflow/core/util/sparse/
sparse_tensor.h
548
const int
slice_dim
= values_inserted_in_slice[slice_index]++;
local
549
output_values_t[slice_index](
slice_dim
) = input_values_t(i);
552
output_indices_t[slice_index](
slice_dim
, j) =
Completed in 100 milliseconds