- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for wrap (0.16 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
s, tracing_ctx->AddParameter(static_cast<DataType>(dtype), partial_shape, &t)); return wrap(t); } void TF_DeleteExecutionContext(TF_ExecutionContext* c) { unwrap(c)->Release(); } TF_AbstractOp* TF_NewAbstractOp(TF_ExecutionContext* c) { return wrap((unwrap(c)->CreateOperation())); } void TF_DeleteAbstractOp(TF_AbstractOp* op) { unwrap(op)->Release(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_eager.cc
using tensorflow::string; using tensorflow::unwrap; using tensorflow::wrap; using tensorflow::strings::StrCat; TF_ExecutionContext* TF_NewEagerExecutionContext(TFE_ContextOptions* options, TF_Status* s) { TFE_Context* c_ctx = TFE_NewContext(options, s); if (TF_GetCode(s) != TF_OK) { return nullptr; } return wrap(static_cast<AbstractContext*>(unwrap(c_ctx))); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jun 25 04:40:46 GMT 2020 - 3.2K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
if (TF_GetCode(status) != TF_OK) return; } std::vector<TFE_TensorHandle*> unwrapped_results(expected_max_outputs); int real_num_outputs = expected_max_outputs; TFE_OpSetCancellationManager(op_.get(), wrap(cancellation_manager_), status); if (TF_GetCode(status) != TF_OK) return; // unwrap op_ and set step_id only if valid step id value was set. // Currently only required for non-TFRT use cases, e.g., EagerOp.
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) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Aug 31 12:04:23 GMT 2020 - 24.9K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
tensorflow::Tensor tensor(dtype, tensorflow::TensorShape({})); std::memcpy(tensorflow::TensorCApi::Buffer(tensor)->data(), data, len); status->status = absl::OkStatus(); return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor)); } // Set server_def on the context, possibly updating it. TF_CAPI_EXPORT extern void TFE_EnableCollectiveOps(TFE_Context* ctx,
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/eager/c_api.cc
return device_.shall_pin_to_this_device(tensorflow::wrap(op), &status); } return errors::Unimplemented("No custom device pinning implementation."); } private: TFE_Context* context_; TFE_CustomDevice device_; void* info_; string name_; }; // An adapter which wraps the shape/data produced by C custom devices and uses // it to implement custom device methods.
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) -
tensorflow/c/eager/dlpack.cc
} break; default: return tensorflow::errors::InvalidArgument("Unsupported Type Codes: ", dtype.code); } } // Wraps the deleter function of DLManagedTensor to match the function signature // TFE_NewTensorHandleFromDeviceMemory. void DeallocatorWrapperFunc(void* data, size_t len, void* dlmt_vptr) { TFE_CallDLManagedTensorDeleter(dlmt_vptr);
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/c_api_experimental.cc
} void TFE_ContextOptionsSetTfrt(TFE_ContextOptions* options, bool use_tfrt) { options->use_tfrt = use_tfrt; } TFE_CancellationManager* TFE_NewCancellationManager() { return tensorflow::wrap(new tensorflow::CancellationManager); } void TFE_CancellationManagerStartCancel( TFE_CancellationManager* cancellation_manager) { tensorflow::unwrap(cancellation_manager)->StartCancel(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/eager/unified_api_testutil.cc
std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus); TFE_TensorHandle* result_t = TF_AbstractTensorGetEagerTensor(wrap(t), status.get()); TF_RETURN_IF_ERROR(StatusFromTF_Status(status.get())); *result_tensor = TFE_TensorHandleResolve(result_t, status.get()); return StatusFromTF_Status(status.get()); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
class GraphContext; class GraphOperation; class GraphTensor; auto& kUnknownDim = shape_inference::InferenceContext::kUnknownDim; auto& kUnknownRank = shape_inference::InferenceContext::kUnknownRank; // GraphTensor wraps a `TF_Output`, i.e. a pointer to TF_Operation and the index // into the list of outputs for the operation. class GraphTensor : public TracingTensorHandle { public: explicit GraphTensor(TF_Output output, TF_Graph* graph)
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1)