- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 21 for X_dims (2.24 sec)
-
tensorflow/c/eager/gradients.cc
num_dims_i, " dimensions which is over the limit of ", TensorShape::MaxDimensions(), ".")); } if (num_dims_i < 0) { proto[i].set_unknown_rank(true); } else { const int64_t* dims_i = dims[i]; auto proto_i = &proto[i]; for (int d = 0; d < num_dims_i; ++d) { proto_i->add_dim()->set_size(dims_i[d]); } } }
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.cc
const int64_t* dims, int num_dims) { DCHECK(index >= 0 && index < shape_list->num_items); TF_ShapeAndType& shape = shape_list->items[index]; DCHECK(shape.dims == nullptr) << "Shape at " << index << " is already set!"; DCHECK(num_dims >= 0) << "Number of dimensions cannot be negative!"; shape.num_dims = num_dims; shape.dims = new int64_t[num_dims];
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/c_test_util.cc
} TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values) { int64_t num_values = 1; for (int i = 0; i < num_dims; ++i) { num_values *= dims[i]; } TF_Tensor* t = TF_AllocateTensor(TF_INT8, dims, num_dims, sizeof(char) * num_values); memcpy(TF_TensorData(t), values, sizeof(char) * num_values); return t; } TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/experimental/gradients/math_grad_test.cc
GTEST_SKIP(); float A_vals[] = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f}; int64_t A_dims[] = {3, 3}; AbstractTensorHandlePtr A; { AbstractTensorHandle* A_raw; status_ = TestTensorHandleWithDims<float, TF_FLOAT>( immediate_execution_ctx_.get(), A_vals, A_dims, 2, &A_raw); ASSERT_EQ(errors::OK, status_.code()) << status_.message(); A.reset(A_raw); }
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
return nullptr; } size_t total_bytes = dl_tensor->dtype.bits / 8; for (int i = 0; i < num_dims; i++) { total_bytes *= dims[i]; } if (dl_tensor->strides != nullptr && !IsValidStrideCompactRowMajorData(dl_tensor->shape, dl_tensor->strides, num_dims)) { status->status = tensorflow::errors::InvalidArgument( "Invalid strides array from DLPack");
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_test.cc
*/ // Build an abstract input tensor. int64_t dims[] = {2, 2}; // Matrices will be 2 x 2 int num_dims = sizeof(dims) / sizeof(dims[0]); float vals[] = {0.0f, 0.0f, 0.0f, 0.0f}; TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); TFE_TensorHandle* t = TestMatrixTensorHandleWithInput(eager_ctx, vals, dims, num_dims); TF_AbstractTensor* at = TF_CreateAbstractTensorFromEagerTensor(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
TF_CAPI_EXPORT extern TF_DataType TF_CheckpointReaderGetVariableDataType( TF_CheckpointReader* reader, const char* name); // Read the shape of a variable and write to `dims` TF_CAPI_EXPORT extern void TF_CheckpointReaderGetVariableShape( TF_CheckpointReader* reader, const char* name, int64_t* dims, int num_dims, TF_Status* status); // Get the number of dimension of a variable TF_CAPI_EXPORT extern int TF_CheckpointReaderGetVariableNumDims(
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/c_api_test_util.cc
constexpr int64_t dims[] = {100, 100}; constexpr int num_elements = dims[0] * dims[1]; float data[num_elements]; for (int i = 0; i < num_elements; ++i) { data[i] = 1.0f; } TF_Status* status = TF_NewStatus(); TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], sizeof(dims) / sizeof(int64_t), status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
DCHECK(shape != nullptr); TF_Status status; int num_dims = TF_GraphGetTensorNumDims(graph_, output_, &status); DCHECK_GE(num_dims, -1); TF_RETURN_IF_ERROR(StatusFromTF_Status(&status)); if (num_dims == kUnknownRank) { return absl::OkStatus(); } std::vector<int64_t> dims(num_dims, kUnknownDim); TF_GraphGetTensorShape(graph_, output_,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
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 = parallel_device.Execute(context.get(), std::vector<ParallelTensor*>(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0)