OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:inner_dim_size
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
segment_reduction_ops_gpu.cu.cc
52
const Index
inner_dim_size
,
58
const Index segment_offset = stripe_index %
inner_dim_size
;
60
stripe_index /
inner_dim_size
* Index(OuterDimTileSize);
78
last_output_segment_id *
inner_dim_size
+ segment_offset;
88
sum += ldg(input + (input_outer_dim_index_base + j) *
inner_dim_size
+
96
last_output_segment_id *
inner_dim_size
+ segment_offset;
103
// 'segment_ids' mapping and '
inner_dim_size
'.
106
const Index
inner_dim_size
,
110
const Index input_total_size = input_outer_dim_size *
inner_dim_size
;
111
const Index output_total_size = output_outer_dim_size *
inner_dim_size
;
[
all
...]
ops_util_test.cc
362
// For multidimensional tensors, alignment is dictated by
inner_dim_size
.
363
int64
inner_dim_size
= EIGEN_MAX_ALIGN_BYTES;
364
Tensor t(DT_FLOAT, TensorShape({3,
inner_dim_size
}));
371
CHECK(sliced.CopyFrom(t.Slice(start, end), TensorShape({1,
inner_dim_size
})));
378
// For multidimensional tensors, alignment is dictated by
inner_dim_size
.
379
int64
inner_dim_size
= EIGEN_MAX_ALIGN_BYTES + 1;
local
380
Tensor t(DT_FLOAT, TensorShape({3,
inner_dim_size
}));
387
CHECK(sliced.CopyFrom(t.Slice(start, end), TensorShape({1,
inner_dim_size
})));
Completed in 72 milliseconds