Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for IntTensorAsVector (0.12 sec)

  1. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc

      // The size of the TF slice operation as a std::vector.  We can always compute
      // this because we only manipulate slices with a Const size.
      std::vector<int64_t> size_as_vector;
    };
    
    std::vector<int64_t> IntTensorAsVector(const Tensor& t) {
      DCHECK(t.dtype() == DT_INT32 || t.dtype() == DT_INT64);
      std::vector<int64_t> result;
      result.reserve(t.NumElements());
      for (int i = 0; i < t.NumElements(); i++) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 13.8K bytes
    - Viewed (0)
Back to top