- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 28 for take (0.2 sec)
-
tensorflow/c/c_api.cc
const char* attr_name, const void* proto, size_t proto_len, TF_Status* status) { // shape.ParseFromArray takes an int as length, this function takes size_t, // make sure there is no information loss. if (proto_len > std::numeric_limits<int>::max()) { status->status = InvalidArgument( "proto_len (", proto_len,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
ParallelTensor::FromTensorHandles(parallel_device, std::move(mixed_handles), status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK); // Can't take the shape of a parallel tensor with varying numbers of axes, but // running operations on them is OK. TF_ASSERT_OK(unknown_length_vector->Shape(&shape)); EXPECT_THAT(*shape, ElementsAre(-1));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
// Update the vector with information from `input_tensors` if provided. if (input_tensors != nullptr) { // Note that we take the address of the elements in `all_input_tensors` // below. Allocate enough space so that no reallocation happens, which will // make the pointers invalid. all_input_tensors.reserve(num_inputs); for (int i = 0; i < num_inputs; ++i) {
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/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/gradients.cc
input_ids[i] = ToId(inputs[i]); input_dtypes[i] = inputs[i]->DataType(); } std::vector<TapeTensor> tape_tensors; tape_tensors.reserve(outputs.size()); for (auto t : outputs) { tape_tensors.push_back(TapeTensor(t)); } GradientTape::RecordOperation( op_name, tape_tensors, input_ids, input_dtypes, [gradient_function]() -> GradientFunction* { return gradient_function; },
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
TFE_TensorHandle* h0 = TestVariable(ctx, 1.0, task1_name); TFE_TensorHandle* h1 = TestVariable(ctx, 2.0, task2_name); TFE_TensorHandle* h2 = TestVariable(ctx, 3.0, task0_name); // Add a sync point to make sure that variables have been initialized // before the function execution starts. TFE_ContextAsyncWait(ctx, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
CHECK_OK(buffer.status()); return new PJRT_Buffer{std::move(*buffer), c_api_client->pjrt_c_client()}; } TEST(TensorPjRtBufferUtilTest, GetPjRtCBufferFromTensorNoBuffer) { auto allocator = std::make_unique<AsyncValueAllocator>(); tensorflow::Tensor tensor(allocator.get(), DT_FLOAT, {1}); EXPECT_THAT( GetPjRtCBufferFromTensor(&tensor), StatusIs(error::INTERNAL, HasSubstr(absl::StrCat(
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler_test.cc
std::unique_ptr<SingleMachine> cluster(new SingleMachine(5 * 60, 3, 0)); TF_ASSERT_OK(cluster->Provision()); TrivialTestGraphInputYielder fake_input(4, 1, 10, false, cluster->GetDeviceNames()); GrapplerItem item; CHECK(fake_input.NextItem(&item)); TF_Status* status = TF_NewStatus(); TF_GraphProperties* graph_properties =
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Thu Apr 13 22:30:58 GMT 2023 - 11.6K bytes - Viewed (0)