Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for UnpackTupleResults (0.28 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.h

                                           llvm::SmallVector<Value>& outputs);
    
      // Given a translated function with a single return value, unpack the tuple
      // results.
      mlir::LogicalResult UnpackTupleResults(mhlo::TupleOp tuple_result,
                                             llvm::SmallVector<Value>& outputs);
    
      // Tries to legalize the specified TensorFlow op, if supported.
      //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

    // multiple values. We get around this by returning a tuple as an XLA op. We
    // then unpack it here to return the multiple values instead.
    mlir::LogicalResult Tf2XlaRewriter::UnpackTupleResults(
        mhlo::TupleOp tuple_result, llvm::SmallVector<Value>& outputs) {
      if (tuple_result->getNumOperands() != op_->getNumResults()) {
        return op_->emitRemark() << "Translated TF2XLA tuple has different "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
Back to top