HomeSort by relevance Sort by last modified time
    Searched refs:contract_pairs (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
blas_gemm.h 60 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; local
61 contract_pairs[0] =
64 c.device(d) = a.contract(b, contract_pairs);
66 c.device(d) += a.contract(b, contract_pairs);
68 c.device(d) = c.constant(alpha) * a.contract(b, contract_pairs) +
  /external/tensorflow/tensorflow/core/kernels/
batch_matmul_op_impl.h 86 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; local
87 contract_pairs[0] = ContractionDims(adj_x, adj_y);
94 z.device(d) = x.contract(y, contract_pairs);
97 z.device(d) = x.contract(y, contract_pairs);
116 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; local
117 contract_pairs[0] = ContractionDims(adj_x, adj_y);
123 z.device(d) = x.contract(y, contract_pairs);
416 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; local
417 contract_pairs[0] = ContractionDims(adj_x, adj_y);
423 z.device(d) = x.contract(y, contract_pairs);
    [all...]

Completed in 69 milliseconds