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 78 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; local
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/
batch_matmul_op_impl.h 88 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; local
89 contract_pairs[0] = ContractionDims(adj_x, adj_y);
96 z.device(d) = x.contract(y, contract_pairs);
99 z.device(d) = x.contract(y, contract_pairs);
118 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; local
119 contract_pairs[0] = ContractionDims(adj_x, adj_y);
125 z.device(d) = x.contract(y, contract_pairs);
508 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; local
509 contract_pairs[0] = ContractionDims(adj_x, adj_y);
515 z.device(d) = x.contract(y, contract_pairs);
    [all...]

Completed in 1192 milliseconds