- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for downTo (0.17 sec)
-
tensorflow/c/c_api.cc
tensorflow::TensorProto from_tensor_proto; status->status = BufferToMessage(from, &from_tensor_proto); if (!status->status.ok()) { return; } status->status = tensorflow::down_cast<tensorflow::TensorInterface*>(to->tensor) ->FromProto(from_tensor_proto); } // --------------------------------------------------------------------------
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/c_api_cluster_test.cc
TFE_ExecutorWaitForAllPendingNodes(executor, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); // TODO(b/136478427): Figure out how to correctly shut the server down. worker_server.release(); // Update the server def with a new set of names (worker instead of // localhost). tensorflow::ServerDef updated_server_def = GetServerDef("worker", 2);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_DeleteExecutor(executor); TFE_DeleteContext(ctx); TF_DeleteStatus(status); // TODO(b/136478427): Figure out how to correctly shut the server down. worker_server.release(); } TEST(CAPI, RemoteExecute) { TestRemoteExecute(false); } TEST(CAPI, RemoteExecuteAsync) { TestRemoteExecute(true); } void TestRemoteExecuteSilentCopiesOp(bool async, bool remote,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 12 00:14:22 GMT 2020 - 5.4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
if (!status.ok()) { CHECK_OK(pjrt::SetPjrtApi(DEVICE_CPU, GetPjrtApi())); } auto pjrt_client = xla::GetCApiClient(DEVICE_CPU); CHECK_OK(pjrt_client.status()); auto c_api_client = down_cast<xla::PjRtCApiClient*>(pjrt_client->get()); std::vector<int32_t> data(1, 0); xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); auto buffer = c_api_client->pjrt_c_client()->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/c_api_distributed_test.cc
TFE_ContextRemoveFunction(ctx, "AddVariablesFunction", status); TFE_DeleteContext(ctx); TF_DeleteStatus(status); // TODO(b/136478427): Figure out how to correctly shut the server down. worker_server1.release(); worker_server2.release(); } TEST(CAPI, TestLocalFunctionWithPackedInput) { TestFunctionWithPackedInput(/*remote=*/false); }
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/eager/c_api_remote_test_util.cc
if (func) { TFE_ContextRemoveFunction(ctx, "MatMulFunction", status); } TFE_DeleteContext(ctx); TF_DeleteStatus(status); // TODO(b/136478427): Figure out how to correctly shut the server down. worker_server1.release(); worker_server2.release();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Dec 11 22:56:03 GMT 2020 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
// One of the inputs we're trying to pack is on a custom device. We'll let // the first custom device we see handle all of the packing. auto* custom_device_handle = tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle); tensorflow::ImmediateExecutionTensorHandle* result; status->status = custom_device_handle->device()->Pack(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
if (!t) { return tensorflow::errors::InvalidArgument( "Unable to cast input to GraphTensor"); } inputs_.push_back(t->output_); *output = tensorflow::down_cast<TracingTensorHandle*>(outputs[0]); return absl::OkStatus(); } Status Finalize(OutputList* outputs, AbstractFunction** f) override { std::vector<TF_Output> graph_outputs;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api.cc
tensorflow::unwrap(h); // TODO(b/175427838): It would be nice to be able to use tensorflow::isa here. if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { return tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle) ->DevicePointer(); } // TODO(b/175427838): It would be nice to be able to use tensorflow::isa here.
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)