| /external/tensorflow/tensorflow/core/kernels/ |
| gemm_functors.h | 98 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
99 dim_pair[0].first = 1;
100 dim_pair[0].second = 0;
102 a_matrix.contract(b_matrix, dim_pair);
|
| mkl_matmul_op.cc | 55 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; variable
56 dim_pair[0].first = transpose_a_ ? 0 : 1;
57 dim_pair[0].second = transpose_b_ ? 1 : 0;
60 ctx, a.dim_size(dim_pair[0].first) == b.dim_size(dim_pair[0].second),
64 int a_dim_remaining = 1 - dim_pair[0].first;
65 int b_dim_remaining = 1 - dim_pair[0].second;
86 const int m = a.dim_size(1 - dim_pair[0].first);
87 const int k = a.dim_size(dim_pair[0].first);
88 const int n = b.dim_size(1 - dim_pair[0].second)
[all...] |
| quantized_matmul_op.cc | 101 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; variable
102 dim_pair[0].first = transpose_a_ ? 0 : 1;
103 dim_pair[0].second = transpose_b_ ? 1 : 0;
106 a.dim_size(dim_pair[0].first) == b.dim_size(dim_pair[0].second),
115 int a_dim_remaining = 1 - dim_pair[0].first;
116 int b_dim_remaining = 1 - dim_pair[0].second;
130 const size_t k = a.dim_size(dim_pair[0].first);
|
| matmul_op.cc | 77 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
80 if (dim_pair[0].second == 0) {
94 if (dim_pair[0].first == 0) {
107 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
122 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
127 bool was_vector = ExplicitVectorMatrixOptimization<T>(a, b, dim_pair, out);
132 b.matrix<T>(), dim_pair);
260 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
271 const uint64 m = a.dim_size(1 - dim_pair[0].first);
272 const uint64 k = a.dim_size(dim_pair[0].first)
465 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; variable
[all...] |
| conv_ops.cc | 80 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
81 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
87 dim_pair);
96 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
97 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
102 filter.shaped<T, 2>({k, filter.dim_size(3)}), dim_pair);
[all...] |
| sparse_matmul_op.cc | 1007 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; variable
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| multithreaded_conv.h | 93 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) {
94 out.device(d) = in0.contract(in1, dim_pair);
130 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
131 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
136 filter, dim_pair);
143 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
144 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
149 filter, dim_pair);
|
