Searched refs:adj_x_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_matmul_op.cc | 28 OP_REQUIRES_OK(ctx, ctx->GetAttr("adj_x", &adj_x_)); in BatchMatMulOp()
33 auto result = xla::BatchDot(MaybeConjugate(ctx->Input(0), adj_x_), adj_x_, in Compile()
39 bool adj_x_; member in tensorflow::__anon748058cc0111::BatchMatMulOp
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 55 OP_REQUIRES_OK(context, context->GetAttr("adj_x", &adj_x_)); in BatchMatMulMkl()
111 if (adj_x_) std::swap(lhs_rows, lhs_cols); in Compute()
117 rhs.shape().DebugString(), " ", adj_x_, " ", adj_y_)); in Compute()
153 bool adj_x_; member in tensorflow::BatchMatMulMkl
207 if (adj_x_) { in CreateMatMulParams()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 653 adj_x_ = false;
656 OP_REQUIRES_OK(context, context->GetAttr("adj_x", &adj_x_));
700 if (adj_x_ || trans_x_) std::swap(d0, d1);
706 " ", adj_x_, " ", adj_y_));
745 ctx, in0_reshaped_float, in1_reshaped_float, adj_x_, adj_y_, trans_x_,
751 adj_x_, adj_y_, trans_x_,
762 bool adj_x_;
|