Home
last modified time | relevance | path

Searched defs:batch_stride (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_unpooling_2d.cc39 const int batch_stride = in MaxUnpooling() local
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEGEMMAssemblyDispatch.cpp365 const size_t batch_stride = batch_size / sizeof(TypeInput); in prepare_indirect_buffer() local
442 const size_t batch_stride = batch_size / sizeof(TypeInputPtr); in configure_indirect() local
/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
Donedevice_train_palm_detection_filter_inference.cc599 const int32_t batch_stride = input_shape[1] * y_stride; in Im2Row() local
Donedevice_train_palm_detection_filter_inference_beta.cc598 const int32_t batch_stride = input_shape[1] * y_stride; in Im2Row() local
Donedevice_train_palm_detection_filter_inference_v2.cc598 const int32_t batch_stride = input_shape[1] * y_stride; in Im2Row() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc1625 int batch_count, uint64_t batch_stride, in ReorganizeMemory()
1671 int batch_count, uint64_t batch_stride, ScratchAllocator *scratch_allocator, in AllocateStridedBuffer()