Searched refs:image_size_vec (Results 1 – 1 of 1) sorted by relevance
398 auto image_size_vec = image_size.vec<int32>(); in ComputeAsync() local399 const int batch_size = internal::SubtleMustCopy(image_size_vec(0)); in ComputeAsync()400 const int image_height = internal::SubtleMustCopy(image_size_vec(1)); in ComputeAsync()401 const int image_width = internal::SubtleMustCopy(image_size_vec(2)); in ComputeAsync()402 const int depth = internal::SubtleMustCopy(image_size_vec(3)); in ComputeAsync()