| /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 | 97 const int dim_pair[] = {1, transpose_b_ ? 1 : 0}; in Compute() local
98 const int batch = src_tf_shape.dim_size(1 - dim_pair[0]); in Compute()
99 const int k = src_tf_shape.dim_size(dim_pair[0]); in Compute()
100 const int channel = weight_tf_shape.dim_size(1 - dim_pair[1]); in Compute()
103 ctx, k == weight_tf_shape.dim_size(dim_pair[1]), in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_fused.cc | 71 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
80 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in operator ()()
103 out.device(d) = lhs.contract(rhs, dim_pair, output_kernel_wrapper); in operator ()()
240 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in operator ()()
273 bool trans_a = dim_pair[0].first == 0 ? true : false; in operator ()()
274 bool trans_b = dim_pair[0].second == 1 ? true : false; in operator ()()
373 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
374 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
375 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
378 ctx, a.dim_size(dim_pair[0].first) == b.dim_size(dim_pair[0].second), in Compute()
[all …]
|
| D | quantized_matmul_op.cc | 116 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
117 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
118 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
121 a.dim_size(dim_pair[0].first) == b.dim_size(dim_pair[0].second), in Compute()
130 int a_dim_remaining = 1 - dim_pair[0].first; in Compute()
131 int b_dim_remaining = 1 - dim_pair[0].second; in Compute()
145 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 | 128 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
129 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
135 dim_pair, std::move(output_kernel_wrapper)); in operator()
145 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
146 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
151 filter.shaped<T, 2>({k, filter.dim_size(3)}), dim_pair, in operator()
|
| D | conv_ops.cc | 106 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
107 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator ()()
113 dim_pair); in operator ()()
122 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
123 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator ()()
128 filter.shaped<T, 2>({k, filter.dim_size(3)}), dim_pair); in operator ()()
1185 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, \
|
| D | conv_2d.h | 289 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
291 out.device(d) = in0.contract(in1, dim_pair, output_kernel);
|
| D | sparse_matmul_op.cc | 1019 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
1020 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
1021 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
1024 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_arith_ops_folder.cc | 41 for (const auto &dim_pair : llvm::enumerate(dims_attr)) { in VerifyReductionInputAndDims() local
42 int64_t cur_dim = dim_pair.value().getSExtValue(); in VerifyReductionInputAndDims()
45 << dim_pair.index() << "-th dimension should be in the range of [-" in VerifyReductionInputAndDims()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 4308 common_factors, [&](const std::pair<int64_t, int64_t>& dim_pair) { in HandleReshape() argument
4309 return dim_pair.first == iota->iota_dimension() && in HandleReshape()
4310 reshape->shape().dimensions(dim_pair.second) > 1; in HandleReshape()
4313 common_factors, [&](const std::pair<int64_t, int64_t>& dim_pair) { in HandleReshape() argument
4314 return dim_pair.first == iota->iota_dimension() + 1; in HandleReshape()
5455 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
5456 arg_dim_unmodified[dim_pair.second] = true; in HandleReduce()
5470 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
5471 if (arg_dim_in_output[dim_pair.second]) { in HandleReduce()
5472 dimensions_not_to_reduce.insert(dim_pair.first); in HandleReduce()
|