- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for pcall (0.18 sec)
-
tensorflow/c/c_api.cc
const void** buf, size_t* len); // TODO(josh11b,mrry): Change Session to be able to use a Graph* // directly, instead of requiring us to serialize to a GraphDef and // call Session::Extend(). bool ExtendSessionGraphHelper(TF_Session* session, TF_Status* status) { if (session->graph != nullptr) { // Take the graph lock before the session lock to avoid deadlock. This is
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
/*dep=*/{"assert"}}, }); TF_ASSERT_OK(ContextFromInterface(unwrap(context.get())) ->AddFunctionDef(assert_and_collective)); std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> call_op( TFE_NewOp(context.get(), "AssertAndCollective", status.get()), TFE_DeleteOp); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); std::unique_ptr<ParallelTensor> reduced_values =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
CheckRemoteMatMulExecutesOK(ctx, remote_device_name, local_device_name); // Adding a non-existent remote worker to cluster def. This should cause the // UpdateServerDef call to fail. tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); tensorflow::JobDef* job_def = cluster_def->mutable_job(0); int port = tensorflow::testing::PickUnusedPortOrDie();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
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) -
tensorflow/c/c_test_util.cc
return TF_NewTensor(TF_FLOAT, nullptr, 0, values, num_bytes, &FloatDeallocator, nullptr); } // All the *Helper methods are used as a workaround for the restrictions that // one cannot call ASSERT_* methods in non-void-returning functions (when // exceptions are disabled during compilation) void PlaceholderHelper(TF_Graph* graph, TF_Status* s, const char* name,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (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) -
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) -
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) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// The resulting TensorHandle owns an opaque pointer to "device memory", which // for a ParallelDevice is really a ParallelTensor. When the TensorHandle is // deleted, it will call ParallelTensorDeallocator to free the struct. ParallelTensor* t_released = t.release(); TFE_CustomDeviceTensorHandleMethods handle_methods; handle_methods.num_dims = &ParallelTensorNumDims;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
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)