- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for one (0.24 sec)
-
tensorflow/c/eager/custom_device_test.cc
std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> one( TestScalarTensorHandle(context.get(), 111.f), TFE_DeleteTensorHandle); op.reset(TFE_NewOp(context.get(), "AssignVariableOp", status.get())); TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); TFE_OpAddInput(op.get(), var_handle, status.get()); TFE_OpAddInput(op.get(), one.get(), status.get()); TFE_OpSetDevice(op.get(), name, status.get());
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/c_test_util.cc
values[0] = v; return TF_NewTensor(TF_FLOAT, nullptr, 0, values, num_bytes, &FloatDeallocator, nullptr); } // All the *Helper methods are used as a workaround for the restrictions that // one cannot call ASSERT_* methods in non-void-returning functions (when // exceptions are disabled during compilation) void PlaceholderHelper(TF_Graph* graph, TF_Status* s, const char* name,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); TensorHandlePtr negative_one_cpu(FloatTensorHandle(3., status.get())); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); components[0] = negative_one_cpu.get(); components[1] = negative_one_cpu.get(); TensorHandlePtr first_negative_one = CreatePerDeviceValues( context.get(), components, first_device_name, status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
} // Add a parallel tensor with different values on each device to the variable. { TensorHandlePtr value_one(FloatTensorHandle(3., status.get())); TensorHandlePtr value_two(FloatTensorHandle(-2., status.get())); std::array<TFE_TensorHandle*, 2> components{value_one.get(), value_two.get()}; TensorHandlePtr combined_value =
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/eager/parallel_device/parallel_device_lib.cc
// Outputs: std::vector<TensorHandlePtr> op_outputs_ TF_GUARDED_BY(execution_mutex_); // TF_Status is an incomplete type and so can't be stack allocated. To avoid // unnecessary allocations each Execute call, we keep one heap-allocated // version for the thread. StatusPtr status_ TF_GUARDED_BY(execution_mutex_); const std::string device_; ExecutorPtr executor_ TF_GUARDED_BY(execution_mutex_);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// // TODO(allenl): There are some use-cases that are only supported by copying to // host at the moment (e.g. debug print on a tensor, .numpy(), etc.). We either // need to return something here or address these use-cases one by one. TFE_TensorHandle* CopyTensorFromParallelDevice(TFE_Context* context, TFE_TensorHandle* tensor, const char* target_device_name,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
const char task1_name[] = "/job:localhost/replica:0/task:1/device:CPU:0"; const char task2_name[] = "/job:localhost/replica:0/task:2/device:CPU:0"; // Create one variable per task. 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);
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/filesystem/modular_filesystem_registration.cc
auto core_ops = std::make_unique<T>(); memset(core_ops.get(), 0, sizeof(T)); memcpy(core_ops.get(), plugin_ops, copy_size); return core_ops; } // Registers one filesystem from the plugin. // // Must be called only with `index` a valid index in `info->ops`. static Status RegisterFileSystem(const TF_FilesystemPluginInfo* info, int index) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 07 22:08:43 GMT 2023 - 12.8K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
TF_SetStatus(status, TF_INVALID_ARGUMENT, "GCS path doesn't contain an object name."); } } /// Appends a trailing slash if the name doesn't already have one. static void MaybeAppendSlash(std::string* name) { if (name->empty()) *name = "/"; else if (name->back() != '/') name->push_back('/'); } // A helper function to actually read the data from GCS.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 23 06:55:53 GMT 2023 - 46.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
tensorflow::ImmediateExecutionTensorHandle* unwrapped_handle = tensorflow::unwrap(handles[i]); if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { // 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 =
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)