Searched defs:im2col_bytes (Results 1 – 2 of 2) sorted by relevance
74 size_t im2col_bytes) { in AllocateTemporaryTensorsIfRequired()189 const size_t im2col_bytes = batches * out_depth * out_height * out_width * in Prepare() local
229 bool is_per_channel, KernelType kernel_type, size_t im2col_bytes) { in AllocateTemporaryTensorsIfRequired()456 const size_t im2col_bytes = static_cast<size_t>(batches) * out_height * in Prepare() local