Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for num_elements (0.19 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top