- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for IntTensorAsVector (0.23 sec)
-
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)