- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 19 for CALL (0.12 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/experimental/grappler/grappler.cc
// Step 1: Load symbol for `TF_InitPlugin` void* dso_symbol; TF_RETURN_IF_ERROR( env->GetSymbolFromLibrary(dso_handle, "TF_InitGraph", &dso_symbol)); // Step 2: Call `TF_InitPlugin` auto init_fn = reinterpret_cast<TFInitGraphPluginFn>(dso_symbol); return InitGraphPlugin(init_fn); } absl::Status InitGraphPlugin(TFInitGraphPluginFn init_fn) {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 15K bytes - Viewed (1) -
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/experimental/next_pluggable_device/c_api.cc
struct TF_VariableInfo { TF_VariableInfo() = delete; // TF_VariableInfo is constructed here by TensorFlow, and will be passed to // plugin as a opaque pointer. Plugin will need to call C APIs below to // operate on TF_VariableInfo (such as allocate temp tensor for the `var` held // by the underlying tensorflow::VariableInfo. TF_VariableInfo(int index, const std::string& name, tensorflow::Var* var) {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1)