- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 23 for if (0.11 sec)
-
tensorflow/c/experimental/gradients/nn_grad.cc
// Calculate Grad std::string name = "relu_grad"; TF_RETURN_IF_ERROR(ReluGrad(ctx, upstream_grad, activations, &grad_inputs[0], name.c_str())); return absl::OkStatus(); } ~ReluGradientFunction() override { for (auto output : forward_outputs_) { if (output) { output->Unref(); } } } private:
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/c_api_unified_experimental.cc
TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(func)); if (!tracing_ctx) { tsl::Set_TF_Status_from_Status( s, tensorflow::errors::InvalidArgument( "TF_AddFunctionParameter must be called on a TracingContext.")); return nullptr; } tensorflow::PartialTensorShape partial_shape; if (shape.num_dims != -1) { DCHECK(shape.dim_sizes != nullptr);
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/dlpack_test.cc
EXPECT_EQ(dltensor_out->dtype.lanes, dltensor_in->dtype.lanes); for (int i = 0; i < dltensor_in->ndim; ++i) { EXPECT_EQ(dltensor_out->shape[i], dltensor_in->shape[i]); if (dltensor_out->strides) { if (i == dltensor_in->ndim - 1) { EXPECT_EQ(dltensor_out->strides[i], 1); } else { EXPECT_EQ(dltensor_out->strides[i],
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
} std::vector<AbstractTensorHandle*> temp_outputs(1); AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, grad_registry)); TF_RETURN_IF_ERROR( forward_model(tape_ctx.get(), inputs, absl::MakeSpan(temp_outputs))); TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx, /*targets=*/temp_outputs, /*sources=*/inputs,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/custom_gradient_test.cc
tape.Watch(inputs[0]); // Watch x. AbstractTensorHandle* exp_output; TF_RETURN_IF_ERROR(ops::Exp(ctx, inputs[0], &exp_output, "Exp")); std::unique_ptr<GradientFunction> gradient_function( new PassThroughGradientFunction); tape.RecordOperation(inputs, {exp_output}, gradient_function.release()); TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx, /*targets*/ {exp_output},
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/array_grad.cc
for (int i = 0; i < grad_outputs.size(); i++) { auto grad_input = grad_outputs[i]; // TODO(srbs): Should we add a copy contructor to AbstractTensorHandle // that takes care of this similar to `Tensor`? if (grad_input) { grad_input->Ref(); } grad_inputs[i] = grad_input; } return absl::OkStatus(); } ~IdentityNGradientFunction() override {} }; } // namespace
C++ - Registered: Tue Apr 09 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 1.6K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_tensor_handle.cc
PartialTensorShape shape; std::string shape_string; if (Shape(&shape).ok()) { shape_string = shape.DebugString(); } else { shape_string = "<error computing shape>"; } std::string value_string; if (!SummarizeValue(value_string).ok()) { value_string = "<error computing value>"; } if (value_string.length() > 100) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 2.1K bytes - Viewed (0) -
tensorflow/c/env.cc
::tensorflow::FileStatistics cc_stats; TF_SetStatus(status, TF_OK, ""); ::tensorflow::Status s = ::tensorflow::Env::Default()->Stat(filename, &cc_stats); ::tensorflow::Set_TF_Status_from_Status(status, s); if (s.ok()) { stats->length = cc_stats.length; stats->mtime_nsec = cc_stats.mtime_nsec; stats->is_directory = cc_stats.is_directory; } }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 11 01:20:50 GMT 2021 - 7K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem_static.cc
// Return value will be unused bool StaticallyRegisterLocalFilesystems() { TF_FilesystemPluginInfo info; TF_InitPlugin(&info); Status status = filesystem_registration::RegisterFilesystemPluginImpl(&info); if (!status.ok()) { VLOG(0) << "Static POSIX filesystem could not be registered: " << status; return false; } return true; } // Perform the actual registration
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 1.5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/array_grad_test.cc
absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) { std::vector<AbstractTensorHandle*> temp_outputs(2); TF_RETURN_IF_ERROR( ops::IdentityN(ctx, inputs, absl::MakeSpan(temp_outputs), "IdentityN")); // Although, `ops::IdentityN` returns 2 tensors, the first tensor isn't needed // for computing gradient so we could safely drop it.
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0)