- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 29 for Jake (0.13 sec)
-
tensorflow/c/eager/gradients_test.cc
int num_retvals = 1; std::vector<AbstractTensorHandle*> outputs(1); GradientRegistry registry; s = RegisterGradients(®istry); ASSERT_EQ(errors::OK, s.code()) << s.message(); auto tape = std::make_unique<Tape>(/*persistent=*/false); s = Execute(check_numerics_op.get(), ctx.get(), absl::MakeSpan(outputs), &num_retvals, &forward_op, tape.get(), registry); ASSERT_EQ(errors::OK, s.code()) << s.message();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
ParallelDevice::ParallelDevice(const std::vector<std::string>& devices, bool is_async, int in_flight_nodes_limit) : underlying_devices_(devices), default_cancellation_manager_(absl::make_unique<CancellationManager>()) { device_threads_.reserve(devices.size()); for (int device_index = 0; device_index < devices.size(); ++device_index) { device_threads_.emplace_back(new DeviceThread(
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/eager/parallel_device/parallel_device_testlib.cc
TFE_OpSetAttrString(op.get(), "container", "", 0); // Use the special GUID for no buffer sharing // // TODO(allenl): Should we provide a better API for this? AFAIK this is the // only reasonable way to make variables with no aliasing using the eager C // API. std::string no_sharing = "cd2c89b7-88b7-44c8-ad83-06c2a9158347"; TFE_OpSetAttrString(op.get(), "shared_name", no_sharing.c_str(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
} // Converts `ninputs` and `inputs` into `inputs_tensors` and `input_nodes` and // does various checks while doing so. `input_nodes` will contain the same // information as input_tensors just in a different structure to make // following processing easier. TODO(iga): Simplify this nested structure. Status ProcessInputs( const TF_Graph* fn_body, const char* fn_name, int ninputs, const TF_Output* inputs, std::vector<OutputTensor>* input_tensors,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/eager/parallel_device/parallel_device.cc
TFE_TensorHandleDevicePointer(tensor, status)); if (TF_GetCode(status) != TF_OK) return nullptr; if (parallel_tensor->num_tensors() == 1) { // Copy-off for single-device tensors is allowed to make debugging dynamic // control flow easier. return TFE_TensorHandleCopySharingTensor(parallel_tensor->tensor(0), status); } else { TF_SetStatus(
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/c_api_experimental_test.cc
CheckOutputShapes(matmul_op, /*input_shapes*/ {make_shape({3, 2}), make_shape({2, 4})}, /*input_tensors*/ {}, /*expected_shape*/ make_shape({3, 4})); // Infer shape when second operand has unknown shape. CheckOutputShapes(matmul_op, /*input_shapes*/ {make_shape({3, 2}), unknown_shape()}, /*input_tensors*/ {},
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jan 17 22:27:52 GMT 2023 - 13.1K bytes - Viewed (1) -
tensorflow/c/experimental/gradients/tape/tape_context.cc
: AbstractContext(kTape), parent_ctx_(c), tape_(tape), registry_(registry) { // TODO(srbs): Make AbstractContext ref counted. // parent_ctx_->Ref(); } void TapeContext::Release() { // TODO(srbs): Change to Unref() delete this; } TapeContext::~TapeContext() { // TODO(srbs): Make AbstractContext ref counted. // parent_ctx_->Unref(); } TapeOperation* TapeContext::CreateOperation() {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 1.7K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/modular_filesystem_registration.cc
size_t plugin_size) { if (plugin_ops == nullptr) return nullptr; size_t copy_size = std::min(plugin_size, sizeof(T)); auto core_ops = std::make_unique<T>(); memset(core_ops.get(), 0, sizeof(T)); memcpy(core_ops.get(), plugin_ops, copy_size); return core_ops; } // Registers one filesystem from the plugin. //
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 07 22:08:43 GMT 2023 - 12.8K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
} absl::Status SetPjRtCBufferToTensor(PJRT_Buffer* c_buffer, xla::PjRtCApiClient* c_api_client, Tensor* tensor) { auto buffer = std::make_unique<xla::PjRtCApiBuffer>(c_api_client, c_buffer); tensorflow::AsyncValueTensor* av_tensor = tensorflow::AsyncValueTensor::FromTensor(tensor); if (av_tensor == nullptr) { TF_ASSIGN_OR_RETURN(
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 3.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
/*run_eager_op_as_function=*/opts->run_eager_op_as_function, /*jit_compile_rewrite=*/opts->jit_compile_rewrite); #if !defined(IS_MOBILE_PLATFORM) eager_context->SetDistributedManager( std::make_unique<tensorflow::EagerContextDistributedManager>( eager_context)); #endif // !IS_MOBILE_PLATFORM return tensorflow::wrap(eager_context); } void TFE_DeleteContext(TFE_Context* ctx) {
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)