- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 2,533 for dtype (0.15 sec)
-
tensorflow/c/eager/dlpack_test.cc
EXPECT_EQ(dltensor_out->device.device_type, dltensor_in->device.device_type); EXPECT_EQ(dltensor_out->device.device_id, dltensor_in->device.device_id); EXPECT_EQ(dltensor_out->ndim, dltensor_in->ndim); EXPECT_EQ(dltensor_out->dtype.code, dltensor_in->dtype.code); EXPECT_EQ(dltensor_out->dtype.bits, dltensor_in->dtype.bits); EXPECT_EQ(dltensor_out->dtype.lanes, dltensor_in->dtype.lanes); for (int i = 0; i < dltensor_in->ndim; ++i) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
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/checkpoint_reader.cc
Status status; if (reader_ != nullptr) { status = reader_->GetTensor(name, out_tensor); } else { tensorflow::DataType dtype; tensorflow::TensorShape shape; status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape); if (status.ok()) { out_tensor->reset(new Tensor(dtype, shape)); status = v2_reader_->Lookup(name, out_tensor->get()); if (!status.ok()) out_tensor->reset(); } }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0) -
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/eager/parallel_device/parallel_device_testlib.h
class Variable { public: // Construct a Variable from a resource-dtype TFE_TensorHandle and an // indication of the dtype of the variable's value. // // Note that creating this resource-dtype handle can fail, so `Create` is a // separate static method which returns a status. Variable(TFE_TensorHandle* handle, TF_DataType type) : handle_(handle), type_(type) {}
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/abstract_tensor_handle.h
tensorflow::PartialTensorShape* shape) const = 0; // Returns tensor (full) type. // While there is no immediate plan to deprecate dtype and shape in favor // of only using full type type information, this is a future possibility. // // Note that map_dtype_to_child_of_tensor() from core/framework/types.h // can be used to set a FullTypeDef based on dtype in a derived class if // appropriate.
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Mar 03 00:30:36 GMT 2023 - 3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
absl::Span<const int64_t> shape, const TF_DataType dtype) : device_(device), tensors_(std::move(tensors)), shape_(std::vector<int64_t>(shape.begin(), shape.end())), dtype_(dtype) {} ParallelTensor(const ParallelDevice& device, std::vector<TensorHandlePtr> tensors, const TF_DataType dtype) : device_(device), tensors_(std::move(tensors)),
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
src/cmd/cgo/gcc.go
} } } // Type returns a *Type with the same memory layout as // dtype when used as the type of a variable or a struct field. func (c *typeConv) Type(dtype dwarf.Type, pos token.Pos) *Type { return c.loadType(dtype, pos, "") } // loadType recursively loads the requested dtype and its dependency graph. func (c *typeConv) loadType(dtype dwarf.Type, pos token.Pos, parent string) *Type {
Go - Registered: Tue Apr 23 11:13:09 GMT 2024 - Last Modified: Thu Nov 02 16:43:23 GMT 2023 - 97K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
TF_Status* s) { unwrap(o)->outputs.push_back(unwrap(tensor)); } void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type, TF_Status* s) { tsl::Set_TF_Status_from_Status( s, unwrap(op)->Reset(op_type, /*raw_device_name=*/nullptr)); } void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
} if (TFE_TensorHandleDataType(component.get()) != dtype) { TF_SetStatus(status, TF_INTERNAL, "Components of a ParallelTensor must all have " "the same dtype"); return nullptr; } } return std::unique_ptr<ParallelTensor>( new ParallelTensor(parallel_device, std::move(components), shape, dtype)); }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1)