- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for num_elements (0.19 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize_op_order.cc
auto get_num_elements = [](RankedTensorType tensor) { int num_elements = 1; for (int i = 0; i < tensor.getRank(); ++i) { // Assume dynamic dim size as the dim size one. if (!tensor.isDynamicDim(i)) { num_elements *= tensor.getDimSize(i); } } return num_elements; }; // If the op is the pass-through op with (3x) smaller output, the dequantize
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/cc/experimental/base/tests/tensor_test.cc
EXPECT_EQ(tensor.dtype(), dtype); EXPECT_EQ(*reinterpret_cast<typename TypeParam::type*>(tensor.data()), 42); EXPECT_EQ(tensor.num_bytes(), sizeof(typename TypeParam::type)); EXPECT_EQ(tensor.num_elements(), 1); } template <typename T> class Construct1DTensorTest : public ::testing::Test {}; TYPED_TEST_SUITE(Construct1DTensorTest, SimpleTypes);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 09:56:08 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/cc/experimental/base/tests/tensorhandle_test.cc
EXPECT_EQ(tensor.dtype(), dtype); EXPECT_EQ(*reinterpret_cast<typename TypeParam::type*>(tensor.data()), 42); EXPECT_EQ(tensor.num_bytes(), sizeof(typename TypeParam::type)); EXPECT_EQ(tensor.num_elements(), 1); } template <typename T> class Construct1DTensorHandleTest : public ::testing::Test {}; TYPED_TEST_SUITE(Construct1DTensorHandleTest, SimpleTypes);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 09:56:08 UTC 2024 - 6.9K bytes - Viewed (0)