Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for some_tensor (0.21 sec)

  1. tensorflow/compiler/mlir/lite/experimental/remat/rematerializer_test.cc

        RandomRemat(const int num_operations, const int num_tensors,
                    const int num_uses, std::mt19937& rng) {
          std::uniform_int_distribution<int> some_size_log(0, 16);
          std::uniform_int_distribution<int> some_tensor(0, num_tensors - 1);
          std::uniform_int_distribution<int> some_operation(0, num_operations - 1);
    
          for (int i = 0; i < num_tensors; ++i) {
            AddTensor(SizeT{1} << some_size_log(rng));
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 14 20:57:44 UTC 2023
    - 19.1K bytes
    - Viewed (0)
Back to top