Searched refs:transpose_output_ (Results 1 – 2 of 2) sorted by relevance
36 context->GetAttr("transpose_output", &transpose_output_)); in XlaDequantizeOp()45 xla::Dequantize<uint8>(input, range, mode_, transpose_output_); in Compile()52 bool transpose_output_; member in tensorflow::__anonbe1cc7520111::XlaDequantizeOp
94 OP_REQUIRES_OK(c, c->GetAttr("transpose_output", &transpose_output_)); in CSRMatMulOp()145 bool transpose_output_; member in tensorflow::CSRMatMulOp207 this->transpose_output_, &output, in Compute()221 if (this->transpose_output_) { in Compute()526 if (this->transpose_output_) { in Compute()591 this->transpose_output_); in Compute()594 if (!this->transpose_output_) { in Compute()614 auto c_mat_col_major = (this->transpose_output_) in Compute()678 if (!this->transpose_output_) { in Compute()776 : transpose_output_(transpose_output) {} in CSRSparseMatrixMatMul()[all …]