Home
last modified time | relevance | path

Searched defs:first_batch (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_concat_op_test.cc76 const int first_batch = 2; in TestSmall8Bit() local
141 const int first_batch = 2; in TestSmall32Bit() local
204 const int first_batch = 2; in TestSecondDim8Bit() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_quantized_concat_op_test.cc78 const int first_batch = 2; in TestSmall8Bit() local
143 const int first_batch = 2; in TestSecondDim8Bit() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DPersistentSoftmax.cuh76 int first_batch = (blockDim.y * blockIdx.x + threadIdx.y) * WARP_BATCH; in softmax_warp_forward() local
223 int first_batch = (blockDim.y * blockIdx.x + threadIdx.y) * WARP_BATCH; in softmax_warp_backward() local