- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for take (0.29 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
TensorHandlePtr combined_value = CreatePerDeviceValues( context.get(), in_components, device_name, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); // Loop to make synchronization failures more deterministic for (int i = 0; i < 100; ++i) { TensorHandlePtr multiply_result( Multiply(context.get(), combined_value.get(), combined_value.get(), status.get()));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 22:09:57 GMT 2023 - 6.7K 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 {}
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/gradient_checker.cc
theta_inputs[input_index] = thetaMinus.get(); TF_RETURN_IF_ERROR( RunAndMaybeSum(ctx, forward, theta_inputs, f_outputs, use_function)); AbstractTensorHandlePtr fMinus(f_outputs[0]); // Take Difference of both estimates: (f(theta + eps) - f(theta - eps)). TF_RETURN_IF_ERROR( ops::Sub(ctx, fPlus.get(), fMinus.get(), f_outputs, "sub_top")); AbstractTensorHandlePtr fDiff(f_outputs[0]);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
TFE_TensorHandle* t = TFE_TensorHandleCopyToDevice( tensor, context, dev->underlying_device.c_str(), status); if (TF_GetCode(status) != TF_OK) return nullptr; auto dst = std::make_unique<LoggedTensor>(t); *(dev->arrived_flag) = true; return MakeLoggedTensorHandle(context, dev->device_name, std::move(dst), status); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
absl::Span<AbstractTensorHandle*> outputs) { Tape tape(/*persistent=*/false); tape.Watch(inputs[0]); AbstractTensorHandle* neg_output; TF_RETURN_IF_ERROR(ops::Neg(ctx, inputs[0], &neg_output, "Neg")); tape.RecordOperation(inputs, {neg_output}, nullptr, "Neg"); return tape.ComputeGradient(ctx, /*targets=*/{neg_output},
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/experimental/gradients/nn_grad_test.cc
#include "tensorflow/c/eager/c_api_test_util.h" #include "tensorflow/c/eager/unified_api_testutil.h" #include "tensorflow/c/experimental/gradients/grad_test_helper.h" #include "tensorflow/c/experimental/gradients/tape/tape_context.h" #include "tensorflow/c/experimental/ops/nn_ops.h" #include "tensorflow/c/tf_status_helper.h" #include "tensorflow/core/platform/tensor_float_32_utils.h" #include "tensorflow/core/platform/test.h"
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 8.3K bytes - Viewed (0) -
tensorflow/c/checkpoint_reader.cc
(*var_to_shape_map)[key] = TensorShape(entry.shape()); (*var_to_data_type_map)[key] = DataType(entry.dtype()); } // The returned pointers are owned by the caller. return std::make_pair(std::move(var_to_shape_map), std::move(var_to_data_type_map)); } } // namespace checkpoint
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0)