Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for RemoteCall (0.09 sec)

  1. tensorflow/compiler/jit/xla_device_ops.h

                                                                                   \
      REGISTER_KERNEL_BUILDER(                                                     \
          Name("RemoteCall").Device(DEVICE).HostMemory("target"), RemoteCallOp);   \
                                                                                   \
      REGISTER_KERNEL_BUILDER(                                                     \
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 23 19:28:25 UTC 2021
    - 17.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc

        if (!symbol) return failure();
        auto f = llvm::dyn_cast<mlir::func::FuncOp>(symbol);
    
        if (call.getOperation()->getNumOperands() !=
            f.getBody().front().getNumArguments()) {
          return failure();  // RemoteCall et al
        }
    
        rewriter.startOpModification(f);
        bool changed = false;
        for (auto [call_arg, body_arg] :
             llvm::zip(call.getOperation()->getOperands(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 9.8K bytes
    - Viewed (0)
Back to top