Home
last modified time | relevance | path

Searched defs:batch_begin (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dcsr_sparse_matrix_to_sparse_tensor.cc169 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()
Dmatrix_band_part.cc174 int64_t batch_begin = start / rows; in BandCompute() local
Dsparse_softmax_cross_entropy_with_logits.cc84 size_t batch_begin = index * classes_num; in SparseSoftmaxCrossEntropyWithLogitsMultiOp() local
/third_party/mesa3d/src/gallium/drivers/d3d12/
Dd3d12_resource_state.h83 struct d3d12_resource_state batch_begin, batch_end; member