Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for IsPackedBlockArgument (0.29 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td

        llvm::ArrayRef<BlockArgument> GetPackedBlockArguments();
        llvm::ArrayRef<BlockArgument> GetReplicatedBlockArguments();
        bool IsReplicatedBlockArgument(BlockArgument block_arg);
        bool IsPackedBlockArgument(BlockArgument block_arg);
        unsigned GetReplicaOperandIndexForBlockArgument(BlockArgument block_arg, unsigned replica);
        Value GetReplicaOperandForBlockArgument(BlockArgument block_arg, unsigned replica);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 23 23:53:20 UTC 2024
    - 14.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc

      assert(block_arg.getOwner() == &GetBody());
      return block_arg.getArgNumber() < GetNumReplicatedBlockArguments();
    }
    
    // Checks if a block argument is packed (forwarding a packed input).
    bool ReplicateOp::IsPackedBlockArgument(BlockArgument block_arg) {
      return !IsReplicatedBlockArgument(block_arg);
    }
    
    // Returns the operand index of the operand being forwarded as a
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.4K bytes
    - Viewed (0)
Back to top