| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_matmul_op.cc | 61 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
62 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
63 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
66 ctx, a.dim_size(dim_pair[0].first) == b.dim_size(dim_pair[0].second), in Compute()
70 int a_dim_remaining = 1 - dim_pair[0].first; in Compute()
71 int b_dim_remaining = 1 - dim_pair[0].second; in Compute()
92 const int m = a.dim_size(1 - dim_pair[0].first); in Compute()
93 const int k = a.dim_size(dim_pair[0].first); in Compute()
94 const int n = b.dim_size(1 - dim_pair[0].second); in Compute()
95 bool transpose_a = dim_pair[0].first == 0; in Compute()
[all …]
|
| D | matmul_op.cc | 77 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in ExplicitVectorMatrixOptimization() argument
80 if (dim_pair[0].second == 0) { in ExplicitVectorMatrixOptimization()
94 if (dim_pair[0].first == 0) { in ExplicitVectorMatrixOptimization()
107 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in ExplicitVectorMatrixOptimization() argument
122 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in launch()
127 bool was_vector = ExplicitVectorMatrixOptimization<T>(a, b, dim_pair, out); in launch()
132 b.matrix<T>(), dim_pair); in launch()
255 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in launch()
266 const uint64 m = a.dim_size(1 - dim_pair[0].first); in launch()
267 const uint64 k = a.dim_size(dim_pair[0].first); in launch()
[all …]
|
| D | quantized_matmul_op.cc | 101 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
102 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
103 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
106 a.dim_size(dim_pair[0].first) == b.dim_size(dim_pair[0].second), in Compute()
115 int a_dim_remaining = 1 - dim_pair[0].first; in Compute()
116 int b_dim_remaining = 1 - dim_pair[0].second; in Compute()
130 const size_t k = a.dim_size(dim_pair[0].first); in Compute()
|
| D | gemm_functors.h | 105 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
106 dim_pair[0].first = 1; in operator()
107 dim_pair[0].second = 0; in operator()
109 a_matrix.contract(b_matrix, dim_pair); in operator()
|
| D | matmul_op.h | 44 const DimPair& dim_pair) { in MatMul() argument
45 out.device(d) = in0.contract(in1, dim_pair); in MatMul()
56 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair);
|
| D | conv_ops.cc | 89 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
90 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator ()()
96 dim_pair); in operator ()()
105 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
106 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator ()()
111 filter.shaped<T, 2>({k, filter.dim_size(3)}), dim_pair); in operator ()()
935 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, \
|
| D | conv_ops_fused_impl.h | 267 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
268 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
274 dim_pair, output_kernel); in operator()
284 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
285 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
290 filter.shaped<T, 2>({k, filter.dim_size(3)}), dim_pair, in operator()
|
| D | conv_2d.h | 141 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
143 out.device(d) = in0.contract(in1, dim_pair, output_kernel);
|
| D | sparse_matmul_op.cc | 1015 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
1016 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
1017 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
1020 left->matrix<float>().contract(right->matrix<float>(), dim_pair); in Compute()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 64 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) { in operator()
65 out.device(d) = in0.contract(in1, dim_pair); in operator()
100 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
101 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
107 filter, dim_pair); in operator()
114 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
115 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
120 filter, dim_pair); in operator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 3027 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
3028 arg_dim_unmodified[dim_pair.second] = true; in HandleReduce()
3042 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
3043 if (arg_dim_in_output[dim_pair.second]) { in HandleReduce()
3044 dimensions_not_to_reduce.insert(dim_pair.first); in HandleReduce()
|