Searched refs:output_batch_offset (Results 1 – 1 of 1) sorted by relevance
200 U* const output_batch_offset = output_flat + batch_idx; in operator ()() local219 output_batch_offset[sample_idx * num_batches] = static_cast<U>(0.0); in operator ()()225 output_batch_offset[sample_idx * num_batches] = in operator ()()236 output_batch_offset[sample_idx * num_batches] = in operator ()()249 output_batch_offset[sample_idx * num_batches] = in operator ()()263 output_batch_offset[sample_idx * num_batches] = in operator ()()276 output_batch_offset[sample_idx * num_batches] = static_cast<U>( in operator ()()287 output_batch_offset[sample_idx * num_batches] = static_cast<U>(NAN); in operator ()()