Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for Mccall (0.14 sec)

  1. tensorflow/c/eager/c_api_experimental.h

    //
    // This call may not block for execution of ops enqueued concurrently with this
    // call.
    TF_CAPI_EXPORT extern void TFE_ExecutorWaitForAllPendingNodes(
        TFE_Executor*, TF_Status* status);
    
    // When an error happens, any pending operations are discarded, and newly issued
    // ops return an error. This call clears the error state and re-enables
    // execution of newly issued ops.
    //
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/dlpack.cc

      auto tf_dlm_type = GetDlDataType(data_type, status);
      if (!status->status.ok()) {
        return nullptr;
      }
    
      TensorReference tensor_ref(*tensor);  // This will call buf_->Ref()
      auto* tf_dlm_tensor_ctx = new TfDlManagedTensorCtx(tensor_ref);
      tf_dlm_tensor_ctx->reference = tensor_ref;
    
      DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients.cc

          output_gradients, result, /*build_default_zeros_grads*/ false));
      return absl::OkStatus();
    }
    
    // Helper functions which delegate to `AbstractOperation`, update
    // the state of the ForwardOperation and call the tape as appropriate.
    // These APIs are mainly to facilitate testing and are subject to change.
    namespace internal {
    Status Reset(AbstractOperation* op_, const char* op,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api.cc

    }
    
    // Note: this function looks up a thread local policy. So it should be called in
    // the appropriate client thread. In particular, in async mode, it may not be
    // safe to call this function from the async EagerExecutor threads.
    extern TFE_ContextDevicePlacementPolicy TFE_ContextGetDevicePlacementPolicy(
        TFE_Context* ctx) {
      return static_cast<TFE_ContextDevicePlacementPolicy>(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  5. tensorflow/c/eager/tape.h

          new GradientTape<Gradient, BackwardFunction, TapeTensor>(false));
      AccumulatorCallState& call_state = call_state_.top();
      call_state.backward_tape = tape.get();
      auto pop_backward_tape =
          gtl::MakeCleanup([&call_state] { call_state.backward_tape = nullptr; });
      std::vector<Gradient*> forwardprop_aids;
      std::vector<int64_t> sources;
      std::unordered_set<int64_t> sources_set;
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  6. tensorflow/c/c_api_experimental.cc

      using tensorflow::AttrBuilder::AttrBuilder;
      // The string buffers to make sure that any `attr_name` we pass into
      // `builder->Set()` will outlive the subsequent
      // `TF_AttrBuilderCheckCanRunOnDevice()` call(s) on the same `builder`.
      std::set<std::string> attr_names;
    };
    
    TF_AttrBuilder* TF_NewAttrBuilder(const char* op_name) {
      return new TF_AttrBuilder(op_name);
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_distributed_test.cc

        if (!enabled_) {
          return absl::OkStatus();
        }
        if (first_call_) {
          first_call_ = false;
          return absl::OkStatus();
        }
        return tensorflow::errors::Internal("Graph pass runs for more than once!");
      }
    
     private:
      bool first_call_ = true;
    };
    
    // After the graph pass is registered, it takes effect globally and can affect
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  8. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      //   Outputs:
      std::vector<TensorHandlePtr> op_outputs_ TF_GUARDED_BY(execution_mutex_);
      // TF_Status is an incomplete type and so can't be stack allocated. To avoid
      // unnecessary allocations each Execute call, we keep one heap-allocated
      // version for the thread.
      StatusPtr status_ TF_GUARDED_BY(execution_mutex_);
    
      const std::string device_;
      ExecutorPtr executor_ TF_GUARDED_BY(execution_mutex_);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
Back to top