- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for DataType (0.17 sec)
-
tensorflow/c/eager/dlpack.cc
break; case TF_DataType::TF_INT8: case TF_DataType::TF_INT16: case TF_DataType::TF_INT32: case TF_DataType::TF_INT64: dtype.code = DLDataTypeCode::kDLInt; break; case TF_DataType::TF_UINT8: case TF_DataType::TF_UINT16: case TF_DataType::TF_UINT32: case TF_DataType::TF_UINT64: dtype.code = DLDataTypeCode::kDLUInt; break;
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/c_api_experimental.cc
TF_DataType value) { auto iter = builder->attr_names.insert(attr_name).first; builder->Set(*iter, static_cast<tensorflow::DataType>(value)); } void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder, const char* attr_name, const TF_DataType* values, int num_values) { auto iter = builder->attr_names.insert(attr_name).first;
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/eager/gradients.cc
handle_ = other.handle_; handle_->Ref(); } TapeTensor::~TapeTensor() { handle_->Unref(); } int64_t TapeTensor::GetID() const { return ToId(handle_); } tensorflow::DataType TapeTensor::GetDType() const { return handle_->DataType(); } AbstractTensorHandle* TapeTensor::GetHandle() const { return handle_; } AbstractTensorHandle* TapeTensor::ZerosLike() const { return nullptr; } class TapeVSpace
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
// 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, TF_DataType dtype, const std::vector<int64_t>& dims, TF_Operation** op) { TF_OperationDescription* desc = TF_NewOperation(graph, "Placeholder", name); TF_SetAttrType(desc, "dtype", dtype);
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_lib_test.cc
// Return values {"y: float"}, // Attr def {}, // Nodes { {{"assert"}, "Assert", {"condition", "x"}, {{"T", std::vector<DataType>{DT_FLOAT}}}}, {{"y"}, "CollectiveReduce", {"x"}, {{"T", DT_FLOAT}, {"group_size", static_cast<int>(devices.size())}, {"group_key", 0},
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) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
public: explicit GraphTensor(TF_Output output, TF_Graph* graph) : TracingTensorHandle(kGraph), output_(output), graph_(graph) {} tensorflow::DataType DataType() const override { return static_cast<tensorflow::DataType>(TF_OperationOutputType(output_)); } tensorflow::Status Shape( tensorflow::PartialTensorShape* shape) const override { DCHECK(shape != nullptr); TF_Status status;
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
"TFE_DeleteTensorHandle", tensorflow::profiler::TraceMeLevel::kInfo); if (h) { tensorflow::unwrap(h)->Unref(); } } TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) { return static_cast<TF_DataType>(tensorflow::unwrap(h)->DataType()); } int TFE_TensorHandleNumDims(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) {
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) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
return nullptr; } TFE_TensorHandleGetStatus(components[0].get(), status); if (!status->status.ok()) { return nullptr; } TF_DataType dtype = TFE_TensorHandleDataType(components[0].get()); // Verify that the TensorHandle's shape and dtype match all of the component // shapes and dtypes. for (TensorHandlePtr& component : components) {
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_testlib.cc
// correspond fairly well to the implementation, but testing the C++ directly is // another option. namespace tensorflow { namespace parallel_device { Variable* Variable::Create(TFE_Context* context, TF_DataType type, const int64_t* dims, const int num_dims, const char* device, TF_Status* status) { std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op(
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/c_api_function.cc
#include "tensorflow/core/util/debug_data_dumper.h" using tensorflow::errors::InvalidArgument; namespace tensorflow { namespace { Status ValidateNonRefOutput(const Node* node, int idx) { const DataType& dt = node->output_type(idx); return IsRefType(dt) ? InvalidArgument("Output ", idx, " of node '", node->name(), "' has a reference type ", DataTypeString(dt))
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2)