Searched refs:trans_y (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 79 bool trans_y, const MatMulBCast& bcast, Tensor* out, in Run()
91 contract_pairs[0] = ContractionDims(adj_x || trans_x, adj_y || trans_y); in Run()
124 bool trans_y, const MatMulBCast& bcast, Tensor* out,
129 contract_pairs[0] = ContractionDims(adj_x || trans_x, adj_y || trans_y);
179 bool adj_y, bool trans_x, bool trans_y,
193 if (!adj_y && !trans_y) {
201 if (!adj_y && !trans_y) {
209 if (!adj_y && !trans_y) {
230 bool trans_y, const MatMulBCast& bcast, Tensor* out) {
253 trans_y, bcast, out, batch_size);
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 571 bool trans_y = !swap_free_and_contract[1]; in ContractOperands() local
583 /*adj_y=*/false, trans_x, trans_y, in ContractOperands()
|