- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 487 for dtype (0.02 sec)
-
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/dlpack_test.cc
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) { EXPECT_EQ(dltensor_out->shape[i], dltensor_in->shape[i]); if (dltensor_out->strides) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Fri Jun 30 03:04:46 UTC 2023 - 4.4K 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_test_util.h
unique_tensor_ptr; TF_Tensor* BoolTensor(int32_t v); // Create a tensor with values of type TF_INT8 provided by `values`. TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values); // Create a tensor with values of type TF_INT32 provided by `values`. TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims, const int32_t* values);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 09 01:06:53 UTC 2018 - 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) -
tensorflow/c/eager/immediate_execution_tensor_handle.cc
const char* device_name = DeviceName(&s); if (!s.ok()) { device_name = "<error fetching device name>"; } return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string, ", dtype=", DataType_Name(DataType()), ", device=\"", device_name, "\")"); } absl::Status ImmediateExecutionTensorHandle::SummarizeValue( std::string& summary) const { absl::Status status;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:24:07 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.h
// // Implicit copies onto `device_name` are allowed, replicating the value once // per device in `underlying_devices`. Implicit copies off of the device throw // an error. // // All component tensors must have the same dtype. Currently they must also have // the same shape, although this requirement may be relaxed in the future. // // `device_name` must not name an existing physical or custom device (see
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Jun 04 21:49:16 UTC 2020 - 2.9K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
auto dtype = TFE_TensorHandleDataType(t->tensor); TFE_CustomDeviceTensorHandleMethods handle_methods; handle_methods.num_dims = &LoggedTensorNumDims; handle_methods.dim = &LoggedTensorDim; handle_methods.deallocator = &LoggedTensorDeallocator; return TFE_NewCustomDeviceTensorHandle(context, logging_device_name.c_str(), dtype, t.release(), handle_methods,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Mar 03 20:47:31 UTC 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
TF_DeleteExecutionContext(ctx); return wrap(func); } TF_AbstractTensor* TF_AddFunctionParameter(TF_ExecutionContext* func, TF_DataType dtype, TF_Shape shape, TF_Status* s) { DCHECK_GE(shape.num_dims, -1); TracingTensorHandle* t; TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(func));
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 9K bytes - Viewed (0)