- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 69 for Num (0.19 sec)
-
tensorflow/c/c_api_experimental.cc
TF_ShapeAndTypeList* TF_NewShapeAndTypeList(int num_items) { TF_ShapeAndTypeList* result = new TF_ShapeAndTypeList; result->num_items = num_items; result->items = (num_items == 0) ? nullptr : new TF_ShapeAndType[num_items](); return result; } void TF_ShapeAndTypeListSetShape(TF_ShapeAndTypeList* shape_list, int index, const int64_t* dims, int num_dims) {
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/experimental/gradients/tape/tape_operation.h
const size_t* lengths, int num_values) override; Status SetAttrFloatList(const char* attr_name, const float* values, int num_values) override; Status SetAttrIntList(const char* attr_name, const int64_t* values, int num_values) override; Status SetAttrTypeList(const char* attr_name, const DataType* values, int num_values) override;
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 3.7K bytes - Viewed (1) -
tensorflow/c/c_api_experimental_test.cc
CHECK_EQ(output_shapes->num_items, 1); int num_dims = output_shapes->items[0].num_dims; int64_t* dims = output_shapes->items[0].dims; if (!expected_shape.has_value()) { EXPECT_EQ(num_dims, -1); EXPECT_EQ(dims, nullptr); return; } EXPECT_EQ(num_dims, expected_shape->size()); for (size_t i = 0; i < num_dims; ++i) {
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_remote_test_util.cc
TFE_TensorHandle* packed_handle = nullptr; if (has_packed_input) { int num_replicas = 1; std::vector<TFE_TensorHandle*> packed_handles = {h1_task2}; packed_handle = TFE_CreatePackedTensorHandle(ctx, packed_handles.data(), &num_replicas, status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); } TFE_Op* matmul = nullptr;
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.h
// Custom devices must use TFE_NewCustomDeviceTensorHandle instead. TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromDeviceMemory( TFE_Context* ctx, const char* device_name, TF_DataType, const int64_t* dims, int num_dims, void* data, size_t len, void (*deallocator)(void* data, size_t len, void* arg), void* deallocator_arg, TF_Status* status); // Retrieves the address space (i.e. job, replia, task) of the local host and
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
tensorflow::ServerDef GetServerDef(const std::string& job_name, int num_tasks) { tensorflow::ServerDef server_def; server_def.set_protocol("grpc"); server_def.set_job_name(job_name); server_def.set_task_index(0); tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster(); tensorflow::JobDef* job_def = cluster_def->add_job(); job_def->set_name(job_name); for (int i = 0; i < num_tasks; i++) {
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/c_api_experimental.h
// b) ConfigProto.gpu_options.allow_growth is set to `gpu_memory_allow_growth`. // c) ConfigProto.device_count is set to `num_cpu_devices`. TF_CAPI_EXPORT extern TF_Buffer* TF_CreateConfig( unsigned char enable_xla_compilation, unsigned char gpu_memory_allow_growth, unsigned int num_cpu_devices); // Create a serialized tensorflow.RunOptions proto, where RunOptions.trace_level
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
} TF_Tensor* TFE_AllocateHostTensor(TFE_Context* ctx, TF_DataType dtype, const int64_t* dims, int num_dims, TF_Status* status) { std::vector<int64_t> dimvec(num_dims); for (int i = 0; i < num_dims; ++i) { dimvec[i] = static_cast<int64_t>(dims[i]); } if (ctx == nullptr) {
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/experimental/next_pluggable_device/c_api.h
// 0 for num_options if no options need to be set. You can use // ConvertToPjRtNamedValueList in // tensorflow/compiler/xla/pjrt/c/pjrt_c_api_helpers.h to generate the options. TF_CAPI_EXPORT extern void TF_CreateAndSetPjRtCApiClient( const char* device_type, TF_Status* status, PJRT_NamedValue* create_options, int num_options);
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Dec 20 20:01:06 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
TFE_NewContextOptions(), TFE_DeleteContextOptions); std::unique_ptr<TF_Buffer, decltype(&TF_DeleteBuffer)> config( TF_CreateConfig( /*xla*/ false, /* gpu_memory_allow_growth */ true, /* num_cpu_devices */ 2), TF_DeleteBuffer); TFE_ContextOptionsSetConfig(opts.get(), config->data, config->length, status.get());
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)