Searched defs:batch_end (Results 1 – 8 of 8) sorted by relevance
36 int64_t batch_end = batch_start + single_thread_tile; in Fp32ReduceSumConcatAxisSizeAVX512Fusion() local104 int64_t batch_end = batch_start + single_thread_tile; in Fp32ReduceSumConcatFusion() local
169 auto sharder = [&](int64_t batch_begin, int64_t batch_end) { in ComputeKernel()187 void CSRSparseMatrixToSparseTensorCpuKernel::SpecialCompute(int64_t batch_begin, int64_t batch_end, in SpecialCompute()
137 int64_t batch_end = (total_rows + rows - 1) / rows; in BandCompute() local175 int64_t batch_end = (end + rows - 1) / rows; in BandCompute() local
85 size_t batch_end = batch_begin + classes_num; in SparseSoftmaxCrossEntropyWithLogitsMultiOp() local
83 struct d3d12_resource_state batch_begin, batch_end; member
377 int batch_end; /* 1 = reposition scanhead */ member
2555 void * batch_end; member
4221 void * batch_end; member