- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for assign (0.42 sec)
-
tensorflow/c/c_api.cc
status->status = InvalidArgument("Unparseable RunOptions proto"); return; } if (run_metadata != nullptr && run_metadata->data != nullptr) { status->status = InvalidArgument("Passing non-empty run_metadata is invalid."); return; } RunMetadata run_metadata_proto; result = session->Run(run_options_proto, input_pairs, output_tensor_names,
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/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
tensorflow::Tensor tensor(allocator.get(), DT_FLOAT, {1}); TF_ASSERT_OK_AND_ASSIGN( auto pjrt_client, xla::GetTfrtCpuClient(/*asynchronous=*/true, /*cpu_device_count=*/1)); std::vector<int32_t> data(1, 0); xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); TF_ASSERT_OK_AND_ASSIGN( auto buffer, pjrt_client->BufferFromHostBuffer(
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/eager/custom_device_test.cc
ASSERT_TRUE(executed); auto handle_cleaner = tensorflow::gtl::MakeCleanup( [var_handle]() { TFE_DeleteTensorHandle(var_handle); }); // Assign to the variable, copying to the custom device. std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> one( TestScalarTensorHandle(context.get(), 111.f), TFE_DeleteTensorHandle);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
} void Variable::AssignAdd(TFE_Context* context, TFE_TensorHandle* value, TF_Status* status) { GeneralAssignment("AssignAddVariableOp", context, value, status); } void Variable::Assign(TFE_Context* context, TFE_TensorHandle* value, TF_Status* status) { GeneralAssignment("AssignVariableOp", context, value, status); }
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_test.cc
std::unique_ptr<CSession> expected_csession( new CSession(expected_graph_, s_)); std::vector<TF_Output> grad_outputs_vec; grad_outputs_vec.assign(grad_outputs, grad_outputs + 2); csession->SetOutputs(grad_outputs_vec); csession->Run(s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); TF_Tensor* out0 = csession->output_tensor(0);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
if (av_tensor == nullptr) { TF_ASSIGN_OR_RETURN( *tensor, MakeTensorFromPjRtBuffer(tensor->dtype(), tensor->shape(), std::move(buffer))); } else { av_tensor->SetBuffer(std::move(buffer)); } return absl::OkStatus(); } absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type) {
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/c_api_experimental_test.cc
TEST(CAPI_EXPERIMENTAL, IsStateful) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus); int assign = TF_OpIsStateful("AssignAddVariableOp", status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); EXPECT_EQ(assign, 1); int id = TF_OpIsStateful("Identity", status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); EXPECT_EQ(id, 0);
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/eager/c_api_test_util.cc
TFE_Execute(op, &var_handle, &num_retvals, status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_DeleteOp(op); if (TF_GetCode(status) != TF_OK) return nullptr; CHECK_EQ(1, num_retvals); // Assign 'value' to it. op = TFE_NewOp(ctx, "AssignVariableOp", status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(op, "dtype", TF_FLOAT); TFE_OpAddInput(op, var_handle, status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/c_api_function_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); TF_GraphCopyFunction(host_graph_, func_, grad_func2, s_); EXPECT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(s_)); EXPECT_EQ(string("Cannot assign gradient function 'MyGrad2' to 'MyFunc' " "because it already has gradient function 'MyGrad1'"), string(TF_Message(s_))); TF_DeleteFunction(grad_func1);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 20 22:08:54 GMT 2023 - 63.6K bytes - Viewed (6)