HomeSort by relevance Sort by last modified time
    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