| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op.cc | 57 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
58 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
59 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
61 int d1 = a.dim_size(dim_pair[0].first); in Compute()
62 int d2 = b.dim_size(dim_pair[0].second); in Compute()
67 int a_dim_remaining = 1 - dim_pair[0].first; in Compute()
68 int b_dim_remaining = 1 - dim_pair[0].second; in Compute()
89 const int m = a.dim_size(1 - dim_pair[0].first); in Compute()
90 const int k = a.dim_size(dim_pair[0].first); in Compute()
91 const int n = b.dim_size(1 - dim_pair[0].second); in Compute()
[all …]
|
| D | mkl_matmul_op_fused.cc | 88 const int dim_pair[] = {1, transpose_b_ ? 1 : 0}; in Compute() local
89 const int batch = src_tf_shape.dim_size(1 - dim_pair[0]); in Compute()
90 const int k = src_tf_shape.dim_size(dim_pair[0]); in Compute()
91 const int channel = weight_tf_shape.dim_size(1 - dim_pair[1]); in Compute()
94 ctx, k == weight_tf_shape.dim_size(dim_pair[1]), in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_fused.cc | 55 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
64 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in operator ()()
84 out.device(d) = lhs.contract(rhs, dim_pair, output_kernel_wrapper); in operator ()()
186 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
187 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
188 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
191 ctx, a.dim_size(dim_pair[0].first) == b.dim_size(dim_pair[0].second), in Compute()
195 int a_dim_remaining = 1 - dim_pair[0].first; in Compute()
196 int b_dim_remaining = 1 - dim_pair[0].second; in Compute()
218 launch(ctx, a, b, dim_pair, fused_computation_, fused_computation_args_, in Compute()
|
| 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_fused_impl.h | 126 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
127 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
133 dim_pair, std::move(output_kernel_wrapper)); in operator()
143 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
144 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
149 filter.shaped<T, 2>({k, filter.dim_size(3)}), dim_pair, in operator()
|
| D | conv_ops.cc | 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 dim_pair); in operator ()()
116 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
117 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator ()()
122 filter.shaped<T, 2>({k, filter.dim_size(3)}), dim_pair); in operator ()()
1158 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, \
|
| D | conv_2d.h | 197 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
199 out.device(d) = in0.contract(in1, dim_pair, output_kernel);
|
| D | sparse_matmul_op.cc | 1014 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
1015 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
1016 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
1019 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/mlir/tensorflow/ir/ |
| D | tf_ops_helpers.inc | 535 for (const auto &dim_pair : llvm::enumerate(dims_attr)) {
536 int64_t cur_dim = dim_pair.value().getSExtValue();
539 << dim_pair.index() << "-th dimension should be in the range of [-"
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 3932 common_factors, [&](const std::pair<int64, int64>& dim_pair) { in HandleReshape() argument
3933 return dim_pair.first == iota->iota_dimension() && in HandleReshape()
3934 reshape->shape().dimensions(dim_pair.second) > 1; in HandleReshape()
3937 common_factors, [&](const std::pair<int64, int64>& dim_pair) { in HandleReshape() argument
3938 return dim_pair.first == iota->iota_dimension() + 1; in HandleReshape()
4655 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
4656 arg_dim_unmodified[dim_pair.second] = true; in HandleReduce()
4670 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
4671 if (arg_dim_in_output[dim_pair.second]) { in HandleReduce()
4672 dimensions_not_to_reduce.insert(dim_pair.first); in HandleReduce()
|