| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batch_matmul_op.cc | 26 OP_REQUIRES_OK(ctx, ctx->GetAttr("adj_x", &adj_x_));
32 /*transpose_x=*/adj_x_, /*transpose_y=*/adj_y_,
33 /*conjugate_x=*/adj_x_, /*conjugate_y=*/adj_y_);
39 bool adj_x_; member in class:tensorflow::__anon38724::BatchMatMulOp
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_batch_matmul_op.cc | 51 OP_REQUIRES_OK(context, context->GetAttr("adj_x", &adj_x_));
82 if (adj_x_) std::swap(lhs_rows, lhs_cols);
88 rhs.shape().DebugString(), " ", adj_x_, " ", adj_y_));
105 const uint64 M = lhs_reshaped.dimension(adj_x_ ? 2 : 1);
106 const uint64 K = lhs_reshaped.dimension(adj_x_ ? 1 : 2);
112 std::vector<MKL_INT> lda_array(batch_size, adj_x_ ? M : K);
128 MklCblasGemmBatch(CblasRowMajor, adj_x_, adj_y_, &m_array[0], &n_array[0],
135 bool adj_x_; member in class:tensorflow::BatchMatMulMkl
|
| batch_matmul_op_impl.h | 444 OP_REQUIRES_OK(context, context->GetAttr("adj_x", &adj_x_));
479 if (adj_x_) std::swap(d0, d1);
485 " ", adj_x_, " ", adj_y_));
501 adj_x_, adj_y_, &out_reshaped);
505 bool adj_x_; member in class:tensorflow::BatchMatMul
|
