- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for CreateLocalHandle (0.24 sec)
-
tensorflow/c/experimental/gradients/nn_grad.cc
} auto imm_ctx = dyn_cast<ImmediateExecutionContext>(ctx); AbstractTensorPtr minus_1(imm_ctx->CreateInt32Scalar(-1)); ImmediateTensorHandlePtr dim(imm_ctx->CreateLocalHandle(minus_1.get())); AbstractTensorHandle* expand_dims_outputs; TF_RETURN_IF_ERROR( ops::ExpandDims(ctx, vec, dim.get(), &expand_dims_outputs, "ExpandDims")); TF_RETURN_IF_ERROR(
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) -
tensorflow/c/eager/immediate_execution_context.h
MemoryReleaser memory_releaser, void* memory_releaser_arg) = 0; // Create a handle to wrap and manage a Tensor virtual ImmediateExecutionTensorHandle* CreateLocalHandle( AbstractTensorInterface* t) = 0; // Copy the handle to another device. virtual ImmediateExecutionTensorHandle* CopyTensorHandleToDevice( ImmediateExecutionTensorHandle* handle, const char* device_name,
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
tensorflow::Tensor tensor; status->status = tensorflow::TF_TensorToTensor(t, &tensor); if (!status->status.ok()) return nullptr; return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor)); } void TFE_DeleteTensorHandle(TFE_TensorHandle* h) { if (h == nullptr) return; tensorflow::profiler::TraceMe activity(
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/c_api_experimental.cc
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, const void* proto,
C++ - Registered: Tue Apr 23 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_experimental.cc
} TFE_TensorHandle* TFE_NewTensorHandleFromTensor(TFE_Context* ctx, TF_Tensor* t, TF_Status* status) { return tensorflow::wrap( tensorflow::unwrap(ctx)->CreateLocalHandle(t->tensor)); } TFE_TensorHandle* TFE_CreatePackedTensorHandle(TFE_Context* ctx, TFE_TensorHandle** handles,
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3)