- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 40 for type (0.16 sec)
-
tensorflow/c/eager/dlpack.cc
owner->reference.Unref(); delete owner; } // Converts TF_DATAType to DLPack data type. DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) { DLDataType dtype; dtype.lanes = 1; dtype.bits = TF_DataTypeSize(data_type) * 8; switch (data_type) { case TF_DataType::TF_BOOL: dtype.code = DLDataTypeCode::kDLBool; break; case TF_DataType::TF_HALF: case TF_DataType::TF_FLOAT:
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
#include "tensorflow/core/framework/full_type.pb.h" #include "tensorflow/core/framework/shape_inference.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/lib/llvm_rtti/llvm_rtti.h" #include "tensorflow/core/platform/errors.h" #include "tensorflow/core/platform/strcat.h" #include "tensorflow/core/platform/types.h" using tensorflow::dyn_cast;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/gradients.cc
input_ids[i] = ToId(inputs[i]); input_dtypes[i] = inputs[i]->DataType(); } std::vector<TapeTensor> tape_tensors; tape_tensors.reserve(outputs.size()); for (auto t : outputs) { tape_tensors.push_back(TapeTensor(t)); } GradientTape::RecordOperation( op_name, tape_tensors, input_ids, input_dtypes, [gradient_function]() -> GradientFunction* { return gradient_function; },
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/experimental/grappler/grappler.h
void* ext; // reserved for future use // Graph C API version. int32_t major_version; int32_t minor_version; int32_t patch_version; // Backend device type supported by the optimizer. const char* device_type; TP_OptimizerConfigs* optimizer_configs; // output, set by plugin TP_Optimizer* optimizer; // output, set by plugin } TP_OptimizerRegistrationParams;
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Aug 03 18:08:43 GMT 2022 - 12.5K 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/c_api_experimental_test.cc
TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx); const char* device_type = TFE_TensorHandleDeviceType(hcpu, status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); ASSERT_TRUE(absl::StrContains(device_type, "CPU")) << device_type; int device_id = TFE_TensorHandleDeviceID(hcpu, status.get());
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/c_api_test_util.cc
for (int i = 0; i < num_devices; ++i) { const string dev_type(TF_DeviceListType(devices, i, status.get())); CHECK_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); const string dev_name(TF_DeviceListName(devices, i, status.get())); CHECK_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); if (dev_type == device_type) { *device_name = dev_name;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/c_test_util.cc
} return found_dtype && found_shape; } bool IsScalarConst(const tensorflow::NodeDef& node_def, int v) { if (node_def.op() != "Const" || node_def.name() != "scalar") { return false; } bool found_dtype = false; bool found_value = false; for (const auto& attr : node_def.attr()) { if (attr.first == "dtype") { if (attr.second.type() == tensorflow::DT_INT32) { found_dtype = true;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/experimental/filesystem/modular_filesystem.cc
memset(&option, 0, sizeof(option)); option.name = const_cast<char*>(name.c_str()); TF_Filesystem_Option_Value option_value; memset(&option_value, 0, sizeof(option_value)); option_value.type_tag = TF_Filesystem_Option_Type_Buffer; option_value.num_values = values.size(); std::vector<TF_Filesystem_Option_Value_Union> option_values(values.size()); for (size_t i = 0; i < values.size(); i++) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 23.1K bytes - Viewed (0)