- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for dim (0.21 sec)
-
tensorflow/c/experimental/grappler/grappler_test.cc
TF_ASSERT_OK(s); EXPECT_EQ(DT_FLOAT, in_props.dtype()); EXPECT_FALSE(in_props.shape().unknown_rank()); EXPECT_EQ(2, in_props.shape().dim_size()); EXPECT_EQ(10, in_props.shape().dim(0).size()); EXPECT_EQ(1, in_props.shape().dim(1).size()); for (int i = 0; i < in_props_buf.size(); i++) TF_DeleteBuffer(in_props_buf[i]); } }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Thu Apr 13 22:30:58 GMT 2023 - 11.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
} int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, TF_Status* status) { 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; }
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
int64_t axis_length = combined_shape.dim_size(axis_index); if (axis_length != component_shape.dim_size(axis_index)) { axis_length = -1; } TF_RETURN_IF_ERROR( combined_shape.SetDimWithStatus(axis_index, axis_length)); } } } auto dim_sizes = combined_shape.dim_sizes(); shape_ = std::vector<int64_t>(dim_sizes.begin(), dim_sizes.end()); }
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/c_api.h
TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims( TFE_TensorDebugInfo* debug_info); // Returns the number of elements in dimension `dim_index`. // Tensor representation on device can be transposed from its representation // on host. The data contained in dimension `dim_index` on device // can correspond to the data contained in another dimension in on-host // representation. The dimensions are indexed using the standard TensorFlow
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// dimension of a parallel tensor. int64_t ParallelTensorDim(void* data, int dim_index, TF_Status* status) { const std::vector<int64_t>* shape; Status s = reinterpret_cast<ParallelTensor*>(data)->Shape(&shape); if (!s.ok()) { tsl::Set_TF_Status_from_Status(status, s); return -1; } return (*shape)[dim_index]; } TF_Buffer* ParallelTensorSummarize(void* data, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
status->status = InvalidArgument("Expected rank is ", num_dims, " but actual rank is ", rank); return; } for (int i = 0; i < num_dims; i++) { dims[i] = shape.dim_size(i); } } int TF_CheckpointReaderGetVariableNumDims(TF_CheckpointReader* reader, const char* name) { const auto& m = reader->GetVariableToShapeMap();
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
} TensorShapeProto proto; if (num_dims < 0) { proto.set_unknown_rank(true); } else { for (int d = 0; d < num_dims; ++d) { proto.add_dim()->set_size(dims[d]); } } forward_op_->attrs.Set(attr_name, proto); return op_->SetAttrShape(attr_name, dims, num_dims); } Status SetAttrFunction(AbstractOperation* op_, const char* attr_name,
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_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..
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_context.h
virtual AbstractTensorInterface* CreateBoolScalar(bool value) = 0; // Tensor creation functions virtual AbstractTensorInterface* CreateTensor( DataType dtype, absl::Span<const int64_t> dim_sizes) = 0; typedef void (*MemoryReleaser)(void* data, size_t len, void* arg); // Create a tensor instance from the given data buffer and description.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.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
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0)