Home
last modified time | relevance | path

Searched defs:batch_index (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/XNNPACK/src/
H A Doperator-run.c346 size_t batch_index, in xnn_compute_spmm()
364 size_t batch_index, in xnn_compute_grouped_batch_igemm()
417 size_t batch_index, in xnn_compute_batch_igemm()
468 size_t batch_index, in xnn_compute_grouped_subgemm2d()
506 size_t batch_index, in xnn_compute_subgemm2d()
543 size_t batch_index, in xnn_compute_grouped_subconv2d()
582 size_t batch_index, in xnn_compute_subconv2d()
620 size_t batch_index, in xnn_compute_conv2d_hwc2chw()
642 size_t batch_index, in xnn_compute_dwconv_unipass()
661 size_t batch_index, in xnn_compute_dwconv2d_chw()
[all …]
/aosp_15_r20/external/mesa3d/src/virtio/vulkan/
H A Dvn_queue.c95 uint32_t batch_index) in vn_get_wait_semaphore_count()
111 uint32_t batch_index) in vn_get_signal_semaphore_count()
127 uint32_t batch_index, in vn_get_wait_semaphore()
148 uint32_t batch_index, in vn_get_signal_semaphore()
188 vn_get_cmd_count(struct vn_queue_submission *submit, uint32_t batch_index) in vn_get_cmd_count()
198 vn_get_cmds(struct vn_queue_submission *submit, uint32_t batch_index) in vn_get_cmds()
211 uint32_t batch_index, in vn_get_cmd()
243 uint32_t batch_index, in vn_get_signal_semaphore_counter()
294 uint32_t batch_index, in vn_fix_batch_cmd_count_for_zink_sync()
320 uint32_t batch_index) in vn_fix_device_group_cmd_count()
[all …]
/aosp_15_r20/external/XNNPACK/test/
H A Dresize-bilinear-operator-tester.h244 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxF16() local
329 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxF32() local
411 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxS8() local
494 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxU8() local
580 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNCHWxF32() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLCrop.cpp54 …sor *src, ICLTensor *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in configure()
60 …sor *src, ICLTensor *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in configure()
73 …onst ITensorInfo *output, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in validate()
H A DCLCropResize.cpp40 …Tensor *output, uint32_t crop_box_ind, Coordinates &start, Coordinates &end, uint32_t &batch_index) in configure_crop()
140 uint32_t batch_index; in configure() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_cross_op.cc283 void Update(const int64_t batch_index, const int64_t cross_count, in Update()
312 string Generate(const int64_t batch_index, in Generate()
338 int64_t Generate(const int64_t batch_index, in Generate()
371 int64_t Generate(const int64_t batch_index, in Generate()
403 int64_t batch_index) in ProductIterator()
624 int batch_index) { in CrossCountByBatchIndex()
H A Dragged_cross_op.cc205 void WriteCombination(int64_t batch_index, in WriteCombination()
217 void WriteCombination(int64_t batch_index, in WriteCombination()
236 void NextCombination(int64_t batch_index, in NextCombination()
612 int batch_index) { in CrossCountByBatchIndex()
H A Din_topk_op_gpu.cu.cc45 const int batch_index = i / num_classes; in ComputePredictionMaskKernel() local
H A Dbatch_kernels.cc860 auto batch_index = in OutputBatch() local
922 const auto batch_index = in Compute() local
H A Dtopk_op_gpu.h347 const int batch_index = blockIdx.x; in TopKKernel() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClCrop.cpp35 …o *src, ITensorInfo *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in configure()
44 …, const ITensorInfo *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in validate()
/aosp_15_r20/external/mesa3d/src/intel/vulkan/grl/gpu/
H A Dbinned_sah_shared.h113 uint batch_index; member
127 ushort batch_index; member
145 ushort batch_index; member
172 uint batch_index; member
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClCropKernel.cpp49 …t ITensorInfo *src, ITensorInfo *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, in configure()
90 …, const ITensorInfo *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in validate()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Doperator-run.c270 size_t batch_index, in compute_sum_rows()
545 size_t batch_index, in compute_max_pooling()
587 size_t batch_index, in compute_average_pooling_unipass()
609 size_t batch_index, in compute_average_pooling_multipass()
660 size_t batch_index) { in compute_global_average_pooling_unipass()
678 size_t batch_index) { in compute_global_average_pooling_multipass()
800 size_t batch_index) { in compute_lut_strided()
839 size_t batch_index) { in compute_clamp_strided()
877 size_t batch_index) { in compute_u8softargmax()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_cholesky_op.cc115 for (int64_t batch_index = batch_begin; batch_index < batch_end; in Compute() local
194 for (int64_t batch_index = batch_begin; batch_index < batch_end; in Compute() local
H A Dsparse_ordering_amd_op.cc100 for (int64_t batch_index = batch_begin; batch_index < batch_end; in Compute() local
/aosp_15_r20/external/mesa3d/src/intel/vulkan/grl/
H A Dgrl_structs.h284 uint batch_index; member
298 ushort batch_index; member
316 ushort batch_index; member
343 uint batch_index; member
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dpadding.cc52 const int32_t batch_index = GetTensorBatchDimIndex(num_dims, data_format); in CheckValidPadding() local
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/preprocessing/benchmarks/
H A Dfeature_column_benchmark.py56 def on_predict_batch_begin(self, batch_index, _): argument
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomialTest.cpp129 const int batch_index = kBatchSize * b; in TEST() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DCropResize.cpp124 …ge(const SimpleTensor<T> &src, Coordinates start, Coordinates end, int32_t batch_index, float extr… in crop_image()
/aosp_15_r20/external/mesa3d/src/mesa/main/
H A Dglthread.c154 unsigned batch_index = batch - ctx->GLThread.batches; in glthread_unmarshal_batch() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNECropKernel.cpp303 …uint32_t batch_index = *(reinterpret_cast<int32_t *>(_box_ind->ptr_to_element(Coordinates(_crop… in run() local
/aosp_15_r20/packages/modules/NeuralNetworks/tools/test_generator/test_harness/include_legacy/
DTestHarness.h411 const int batch_index = kBatchSize * b; in expectMultinomialDistributionWithinTolerance() local
/aosp_15_r20/external/mesa3d/src/intel/tools/
H A Dintel_dump_gpu.c214 int batch_index; in dump_execbuffer2() local

12