- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 21 for Type (0.12 sec)
-
tensorflow/c/c_api_experimental.cc
} void TF_AttrBuilderCheckCanRunOnDevice(TF_AttrBuilder* builder, const char* device_type, TF_Status* status) { status->status = tensorflow::FindKernelDef( tensorflow::DeviceType(device_type), builder->BuildNodeDef(), /* def = */ nullptr, /* kernel_class_name = */ nullptr); }
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/filesystem/plugins/posix/posix_filesystem.cc
if (n > INT32_MAX) requested_read_length = INT32_MAX; else requested_read_length = n; // `pread` returns a `ssize_t` on POSIX, but due to interface being // cross-platform, return type of `Read` is `int64_t`. int64_t r = int64_t{pread(posix_file->fd, dst, requested_read_length, static_cast<off_t>(offset))}; if (r > 0) { dst += r;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
std::string device_type = parsed_name.type; int device_id = 0; if (parsed_name.has_id) { device_id = parsed_name.id; } ctx.device_id = device_id; if (device_type == "CPU") { ctx.device_type = DLDeviceType::kDLCPU; } else if (device_type == "GPU") { #if TENSORFLOW_USE_ROCM ctx.device_type = DLDeviceType::kDLROCM; #else
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler.cc
TF_RETURN_IF_ERROR(ValidateTPOptimizerConfigs(optimizer_configs)); CGraphOptimizerRegister( [=]() { return new CGraphOptimizer(optimizer, params.device_type); }, optimizer_configs, params.device_type); return absl::OkStatus(); } } // namespace grappler } // namespace tensorflow void TF_GetNodesToPreserveListSize(const TF_GrapplerItem* item, int* num_values,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 15K bytes - Viewed (1) -
tensorflow/c/eager/gradients.cc
gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) const override; // Calls the passed-in backward function. // op_type is the op's name provided in RecordOperation. Status CallBackwardFunction( const string& op_type, GradientFunction* gradient_function, const std::vector<int64_t>& unneeded_gradients, gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
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/c_test_util.cc
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/custom_device_test.cc
executed = false; TFE_Execute(matmul.get(), &retval, &num_retvals, status.get()); EXPECT_FALSE(executed); ASSERT_FALSE(TF_GetCode(status.get()) == TF_OK); // Custom devices can refuse to do type-based dispatch (as hcustom1 is // configured to do) matmul.reset(MatMulOp(context.get(), hcustom1.get(), hcpu.get())); num_retvals = 1; executed = false;
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/experimental/next_pluggable_device/c_api.cc
tensorflow::DeviceType(device_type), std::move(*pjrt_client)); status->status = s; } void TF_ResetPjRtCClient(const char* device_type, TF_Status* status) { status->status = tensorflow::ResetPjRtClient(tensorflow::DeviceType(device_type)); } PJRT_Client* TF_GetPjRtCClient(const char* device_type, TF_Status* status) { absl::StatusOr<xla::PjRtCApiClient*> pjrt_c_api_client =
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/eager/c_api_distributed_test.cc
" signature {" " name: 'AddVariablesFunction'" " input_arg {" " name: 'var'" " type: DT_RESOURCE" " }" " output_arg {" " name: 'sum'" " type: DT_FLOAT" " }" " }" " node_def {" " name: 'read0'" " op: 'ReadVariableOp'" " input: 'var'"
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/grappler/grappler_test.cc
TF_SetStatus(status, TF_OK, ""); PopulateDefaultParam(params); params->device_type = "Device0"; }; auto plugin_init_1 = [](TP_OptimizerRegistrationParams* const params, TF_Status* const status) -> void { TF_SetStatus(status, TF_OK, ""); PopulateDefaultParam(params); params->device_type = "Device1"; }; TF_ASSERT_OK(InitGraphPlugin(plugin_init_0));
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)