Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/c/experimental/gradients/nn_grad.cc

                        absl::Span<AbstractTensorHandle*> outputs) {
      if (!isa<ImmediateExecutionContext>(ctx)) {
        // TODO(b/168850692): Fix this.
        return errors::Unimplemented(
            "BroadcastMul is not supported in tracing mode yet.");
      }
      auto imm_ctx = dyn_cast<ImmediateExecutionContext>(ctx);
      AbstractTensorPtr minus_1(imm_ctx->CreateInt32Scalar(-1));
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/immediate_execution_distributed_manager.h

    #include <cstdint>
    #include <string>
    
    #include "tensorflow/core/platform/status.h"
    
    namespace tsl {
    class CoordinationServiceAgent;
    }
    
    namespace tensorflow {
    class ImmediateExecutionContext;
    class ServerDef;
    class WorkerEnv;
    class WorkerCacheInterface;
    
    class ImmediateExecutionDistributedManager {
     public:
      virtual ~ImmediateExecutionDistributedManager() {}
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 2.9K bytes
    - Viewed (0)
Back to top