Searched defs:total_batches (Results 1 – 2 of 2) sorted by relevance
144 const size_t total_batches = src->tensor_shape().total_size_upper(3); in configure() local236 const size_t total_batches = window.shape().total_size_upper(3); in run_op() local
162 const size_t total_batches = dst->tensor_shape().total_size_upper(3); in configure() local