- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for dim (0.19 sec)
-
tensorflow/c/c_api_test.cc
#undef EXPECT_TF_META TEST(CAPI, TestTensorAligned) { int64_t dim = 7; size_t tensor_size_bytes = dim * TF_DataTypeSize(TF_FLOAT); TF_Tensor* a = TF_AllocateTensor( /*dtype=*/TF_FLOAT, /*dims=*/&dim, /*num_dims=*/1, /*len=*/tensor_size_bytes); float* data = reinterpret_cast<float*>(TF_TensorData(a)); for (int i = 0; i < dim; ++i) { data[i] = 0; } if (TF_TensorDefaultAlignment() > 0) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api_test.cc
ASSERT_EQ(num_dims, -1); ASSERT_EQ("Invalid handle", string(TF_Message(status.get()))); TF_SetStatus(status.get(), TF_OK, ""); int dim = TFE_TensorHandleDim(h, 0, status.get()); ASSERT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(status.get())); ASSERT_EQ(dim, -1); ASSERT_EQ("Invalid handle", string(TF_Message(status.get()))); } TEST(CAPI, TensorHandleDevices) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/c_api.h
// has registration: // REGISTER_OP("Concat") // .Input("concat_dim: int32") // .Input("values: N * T") // .Output("output: T") // .Attr("N: int >= 2") // .Attr("T: type"); // that defines two inputs, "concat_dim" and "values" (in that order). // You must use TF_AddInput() for the first input (since it takes a
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3)