Home
last modified time | relevance | path

Searched refs:adj_y_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_matmul_op.cc29 OP_REQUIRES_OK(ctx, ctx->GetAttr("adj_y", &adj_y_)); in BatchMatMulOp()
37 MaybeConjugate(ctx->Input(1), adj_y_), adj_y_)); in Compile()
43 bool adj_y_; member in tensorflow::__anon37d5f8050111::BatchMatMulOp
/external/tensorflow/tensorflow/core/kernels/
Dmkl_batch_matmul_op.cc52 OP_REQUIRES_OK(context, context->GetAttr("adj_y", &adj_y_)); in BatchMatMulMkl()
83 if (adj_y_) std::swap(rhs_rows, rhs_cols); in Compute()
88 rhs.shape().DebugString(), " ", adj_x_, " ", adj_y_)); in Compute()
107 const uint64 N = rhs_reshaped.dimension(adj_y_ ? 1 : 2); in Compute()
113 std::vector<MKL_INT> ldb_array(batch_size, adj_y_ ? K : N); in Compute()
128 MklCblasGemmBatch(CblasRowMajor, adj_x_, adj_y_, &m_array[0], &n_array[0], in Compute()
136 bool adj_y_; member in tensorflow::BatchMatMulMkl
Dbatch_matmul_op_impl.h537 OP_REQUIRES_OK(context, context->GetAttr("adj_y", &adj_y_));
572 if (adj_y_) std::swap(d2, d3);
577 " ", adj_x_, " ", adj_y_));
593 adj_x_, adj_y_, &out_reshaped);
598 bool adj_y_;