Searched refs:adj_y_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_matmul_op.cc | 29 OP_REQUIRES_OK(ctx, ctx->GetAttr("adj_y", &adj_y_)); in BatchMatMulOp()
34 MaybeConjugate(ctx->Input(1), adj_y_), adj_y_); in Compile()
40 bool adj_y_; member in tensorflow::__anon748058cc0111::BatchMatMulOp
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 56 OP_REQUIRES_OK(context, context->GetAttr("adj_y", &adj_y_)); in BatchMatMulMkl()
112 if (adj_y_) std::swap(rhs_rows, rhs_cols); in Compute()
117 rhs.shape().DebugString(), " ", adj_x_, " ", adj_y_)); in Compute()
154 bool adj_y_; member in tensorflow::BatchMatMulMkl
217 if (adj_y_) { in CreateMatMulParams()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 654 adj_y_ = false;
657 OP_REQUIRES_OK(context, context->GetAttr("adj_y", &adj_y_));
701 if (adj_y_ || trans_y_) std::swap(d2, d3);
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_,
763 bool adj_y_;
|