Searched defs:tensor_bytes (Results 1 – 2 of 2) sorted by relevance
81 auto tensor_bytes = tensor_size_ * sizeof(T); in LaunchKernel() local
363 auto tensor_bytes = static_cast<size_t>(tensor->data().nbytes()); in GroupingTensors() local462 auto tensor_bytes = static_cast<size_t>(tensor->data().nbytes()); in GetFusionSize() local