H A D | batch_matmul_op_impl.h | 48 Eigen::IndexPair<Eigen::DenseIndex> ContractionDims(bool adj_x, bool adj_y) { argument 50 if (!adj_y) { 56 if (!adj_y) { 75 const Tensor in_y, bool adj_x, bool adj_y, Tensor* out, 87 contract_pairs[0] = ContractionDims(adj_x, adj_y); 92 if (adj_x != adj_y) { 111 const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out, 117 contract_pairs[0] = ContractionDims(adj_x, adj_y); 131 static void Multiply(bool adj_x, bool adj_y, Tx x, Ty y, Tz z) { argument 133 if (!adj_y) { 74 Run(const OpKernelContext* context, const Tensor& in_x, const Tensor in_y, bool adj_x, bool adj_y, Tensor* out, int start, int limit) argument 110 Run(const OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out, int start, int limit) argument 170 Run(const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out, int start, int limit) argument 201 Launch(OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out) argument 290 Launch(OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out) argument 410 Run(const OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out, int start, int limit) argument 430 Launch(OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out) argument [all...] |