Home
last modified time | relevance | path

Searched defs:batch_input_offset (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu267 const int batch_input_offset = offsets[batch_id]; in add_padding_1() local
277 const int batch_input_offset = offsets[batch_id]; in add_padding_1() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dneon_tensor_utils.cc1318 const int batch_input_offset = input_offset[batch]; in NeonMatrixBatchVectorMultiplyAccumulateImpl() local