Searched refs:trans_x (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 78 const Tensor in_y, bool adj_x, bool adj_y, bool trans_x, in Run()
91 contract_pairs[0] = ContractionDims(adj_x || trans_x, adj_y || trans_y); in Run()
123 const Tensor& in_y, bool adj_x, bool adj_y, bool trans_x,
129 contract_pairs[0] = ContractionDims(adj_x || trans_x, adj_y || trans_y);
179 bool adj_y, bool trans_x, bool trans_y,
192 if (!adj_x && !trans_x) {
229 const Tensor& in_y, bool adj_x, bool adj_y, bool trans_x,
252 ParallelMatMulKernel::Run(context, in_x, in_y, adj_x, adj_y, trans_x,
260 [&in_x, &in_y, adj_x, adj_y, trans_x, trans_y, &bcast, out](
263 trans_x, trans_y, bcast, out,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 570 bool trans_x = swap_free_and_contract[0]; in ContractOperands() local
583 /*adj_y=*/false, trans_x, trans_y, in ContractOperands()
|