OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:glimpse_cols
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
eigen_attention_test.cc
34
const ptrdiff_t
glimpse_cols
= 6;
local
46
Tensor<float, 4> result(depth, glimpse_rows,
glimpse_cols
, batch);
47
result = ExtractGlimpses(input, glimpse_rows,
glimpse_cols
, offsets);
50
for (int c = 0; c <
glimpse_cols
; ++c) {
52
c + ((1.0f + offsets[b].second) * cols -
glimpse_cols
) / 2;
70
const ptrdiff_t
glimpse_cols
= 6;
local
82
Tensor<float, 4> result(depth, glimpse_rows,
glimpse_cols
, batch);
83
result = ExtractGlimpses(input, glimpse_rows,
glimpse_cols
, offsets);
86
for (int c = 0; c <
glimpse_cols
; ++c) {
88
c + ((1.0f + offsets[b].second) * cols -
glimpse_cols
) / 2
[
all
...]
/external/tensorflow/tensorflow/python/kernel_tests/
attention_ops_test.py
37
glimpse_sizes: Dimensions of the glimpse in [glimpse_rows,
glimpse_cols
].
83
glimpse_cols
= (array_ops.transpose(
88
value_rows, value_cols = sess.run([glimpse_rows,
glimpse_cols
])
Completed in 88 milliseconds