Searched refs:contract_pairs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | blas_gemm.h | 78 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs;
79 contract_pairs[0] =
83 c.device(d) = a.contract(b, contract_pairs);
86 c.device(d) += a.contract(b, contract_pairs);
88 c.device(d) = c.constant(T(alpha)) * a.contract(b, contract_pairs) +
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 90 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; in Run() local
91 contract_pairs[0] = ContractionDims(adj_x || trans_x, adj_y || trans_y); in Run()
106 z.device(d) = x.contract(y, contract_pairs); in Run()
109 z.device(d) = x.contract(y, contract_pairs); in Run()
128 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs;
129 contract_pairs[0] = ContractionDims(adj_x || trans_x, adj_y || trans_y);
134 Tz.device(d) = Tx.contract(Ty, contract_pairs);
149 z.device(d) = x.contract(y, contract_pairs);
|