- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 431 for dtype (1.3 sec)
-
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)),
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 13.1K 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)); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K 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:
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 12.9K bytes - Viewed (0) -
tensorflow/c/checkpoint_reader.cc
absl::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(); } }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/c/eager/abstract_tensor_handle.h
// 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.
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
void* data, size_t len, TF_Status* status) { auto dtype = static_cast<tensorflow::DataType>(data_type); DCHECK(tensorflow::DataTypeCanUseMemcpy(dtype)); tensorflow::Tensor tensor(dtype, tensorflow::TensorShape({})); std::memcpy(tensorflow::TensorCApi::Buffer(tensor)->data(), data, len); status->status = absl::OkStatus();
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 29.5K 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 {
Registered: Tue Nov 05 11:13:11 UTC 2024 - Last Modified: Wed Sep 18 15:07:34 UTC 2024 - 97.1K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT); TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); auto outputs =
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 15.6K bytes - Viewed (0) -
tensorflow/c/eager/abstract_tensor_handle.cc
shape_string = "<error computing shape>"; } else { shape_string = shape.DebugString(); } return absl::StrCat("TensorHandle(shape=", shape_string, ", dtype=", DataType_Name(DataType()), ", type=", FullType().DebugString(), ")"); } absl::Status AbstractTensorHandle::TensorHandleStatus() const { // Tensor handles in current runtime don't carry error info and this method
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/gradients.h
void Watch(const AbstractTensorHandle*); // Records an operation with given inputs and outputs // on the tape and marks all its outputs as watched if at // least one input of the op is watched and has a trainable dtype. // op_name is optional and is used for debugging only. void RecordOperation(absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle* const> outputs,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 6.9K bytes - Viewed (0)