- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for dim (0.03 sec)
-
tensorflow/c/eager/c_api.cc
if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return -1; } int64_t dim = -1; status->status = tensorflow::unwrap(h)->Dim(dim_index, &dim); return dim; } const char* TFE_TensorHandleDeviceName(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
// Information about the shape of a Tensor and its type. struct TF_ShapeAndType { // Number of dimensions. -1 indicates unknown rank. int num_dims; // Array of dimensions. -1 indicates unknown dim. int64_t* dims; // The data type. May be 0 to denote unknown type. TF_DataType dtype; }; typedef struct TF_ShapeAndType TF_ShapeAndType; // A list of TF_ShapeAndType elements..
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
shape.dims = nullptr; continue; } shape.dims = new int64_t[shape.num_dims]; for (size_t j = 0; j < shape.num_dims; ++j) { shape.dims[j] = c.Value(c.Dim(shape_handle, j)); } } if (output_shapes != nullptr) *output_shapes = output_shapes_result; // TODO(bgogul): Set output_resource_shapes_and_types. }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 29.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
ParallelTensor* t_released = t.release(); TFE_CustomDeviceTensorHandleMethods handle_methods; handle_methods.num_dims = &ParallelTensorNumDims; handle_methods.dim = &ParallelTensorDim; handle_methods.deallocator = &ParallelTensorDeallocator; handle_methods.summarize = &ParallelTensorSummarize; return TensorHandlePtr(TFE_NewCustomDeviceTensorHandle(
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
// shape is explicitly requested where possible. int (*num_dims)(void* data, TF_Status* status); // Computes the axis length at `dim_index`. int64_t (*dim)(void* data, int dim_index, TF_Status* status); void (*deallocator)(void* data); // Summarizes the value of this tensor. The caller takes ownership of the
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 39.5K bytes - Viewed (0)