Searched defs:batch_stride (Results 1 – 12 of 12) sorted by relevance
254 const size_t batch_stride = batch_size / sizeof(TypeInput); in prepare_indirect_buffer() local331 const size_t batch_stride = batch_size / sizeof(TypeInputPtr); in configure_indirect() local
82 int64_t batch_stride; // `batch_stride` is set to `0` when `batch_size == 1`. member
99 int64_t batch_stride; member
130 int64_t batch_stride = num_rows * num_cols; in For() local
39 const int batch_stride = in MaxUnpooling() local
85 auto batch_stride = ndim > 2 && batch_offset >= 0 ? input_strides[ndim - 3] : 0; in createRawDnMatDescriptor() local
194 int64_t batch_stride; member
143 std::optional<int64_t> batch_stride) { in Create()
666 int batch_count, uint64_t batch_stride, in ReorganizeMemory()712 int batch_count, uint64_t batch_stride, ScratchAllocator *scratch_allocator, in AllocateStridedBuffer()
598 const int32_t batch_stride = input_shape[1] * y_stride; in Im2Row() local
599 const int32_t batch_stride = input_shape[1] * y_stride; in Im2Row() local