- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getAdjX (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc
return failure(); } // Replace the last 2 dimensions of LHS and RHS if necessary. // The actual transpose is done by MatMulOp. if (op.getAdjX()) { std::swap(lhs_shape[lhs_dims - 1], lhs_shape[lhs_dims - 2]); } if (op.getAdjY()) { std::swap(rhs_shape[rhs_dims - 1], rhs_shape[rhs_dims - 2]); } const int64_t rows = lhs_shape[lhs_dims - 2];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
Value input_rhs = get_real_input_value(op.getY()); Value legalized_lhs = op.getAdjX() ? create_z_x_transpose_op(input_lhs) : input_lhs; // The rhs need to be transposed if adj_y == false AND this matmul will be // legalized to tfl.fully_connected Value legalized_rhs = !op.getAdjY() ? create_z_x_transpose_op(input_rhs) : input_rhs; Type output_type = op.getResult().getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0)