Home
last modified time | relevance | path

Searched defs:in_batch (Results 1 – 25 of 65) sorted by relevance

123

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/common/
H A Dshims.hpp174 const T* const in_batch = in + n*in_batch_stride; in nchw_to_nhwc() local
345 const T* const in_batch = in + n*in_batch_stride; in nchw_to_nhwc() local
405 const T* const in_batch = in + n*in_batch_stride; in nhwc_to_nchw() local
588 const T* const in_batch = in + n*in_batch_stride; in nhwc_to_nchw() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_op.cc131 const int64_t in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
272 const int64_t in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
373 const int64_t in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
1297 const int64_t in_batch = GetTensorDim(x, tensor_format_, 'N'); in ComputeWithReservedSpace() local
1542 const int64_t in_batch = GetTensorDim(x, tensor_format_, 'N'); in ComputeWithReservedSpace() local
H A Dcudnn_pooling_gpu.cc45 const int64_t in_batch = GetTensorDim(tensor_in, data_format, 'N'); in Compute() local
140 const int64_t in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute() local
H A Dconv_ops_3d.cc149 const int64_t in_batch = GetTensorDim(input, data_format_, 'N'); in Compute() local
240 const int64_t in_batch = GetTensorDim(input, data_format, 'N'); in launch() local
H A Dpooling_ops_3d.cc172 const int64_t in_batch = GetTensorDim(tensor_in, data_format_, 'N'); in Compute() local
367 const int64_t in_batch = GetTensorDim(tensor_in, data_format_, 'N'); in Compute() local
H A Dfractional_avg_pool_op.cc296 const int64_t in_batch = orig_input_tensor_shape_flat(0); in Compute() local
H A Dconv_ops.cc821 const int64_t in_batch = GetTensorDim(input, data_format, 'N'); in operator ()() local
/aosp_15_r20/external/libbpf/src/
H A Dbpf.c523 static int bpf_map_batch_common(int cmd, int fd, void *in_batch, in bpf_map_batch_common()
558 int bpf_map_lookup_batch(int fd, void *in_batch, void *out_batch, void *keys, in bpf_map_lookup_batch()
566 int bpf_map_lookup_and_delete_batch(int fd, void *in_batch, void *out_batch, in bpf_map_lookup_and_delete_batch()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dbatch_to_space_nd.h73 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DRepeat.cpp82 int32_t in_batch = utils::safe_downcast<int32_t>(dim_at<kBatch4D>(in_sizes)); in add_repeat_channel_node() local
/aosp_15_r20/external/executorch/backends/cadence/reference/operators/
H A Dquantized_conv_out.cpp73 const IT* in_batch = p_in + _n * c * h * w; in conv2d_nchw_core_generic() local
/aosp_15_r20/external/bcc/src/cc/
H A Dlibbpf.c435 int bpf_lookup_batch(int fd, __u32 *in_batch, __u32 *out_batch, void *keys, in bpf_lookup_batch()
452 int bpf_lookup_and_delete_batch(int fd, __u32 *in_batch, __u32 *out_batch, in bpf_lookup_and_delete_batch()
/aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
H A Donedevice_train_palm_detection_filter_inference_beta.cc469 for (int32_t in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceNDRank4() local
H A Donedevice_train_palm_detection_filter_inference.cc470 for (int32_t in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceNDRank4() local
H A Donedevice_train_palm_detection_filter_inference_v2.cc469 for (int32_t in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceNDRank4() local
/aosp_15_r20/prebuilts/vndk/v31/arm64/include/bionic/libc/kernel/uapi/linux/
Dbpf.h316 __aligned_u64 in_batch; member
/aosp_15_r20/prebuilts/vndk/v31/arm/include/bionic/libc/kernel/uapi/linux/
Dbpf.h316 __aligned_u64 in_batch; member
/aosp_15_r20/prebuilts/vndk/v31/x86_64/include/bionic/libc/kernel/uapi/linux/
Dbpf.h316 __aligned_u64 in_batch; member
/aosp_15_r20/prebuilts/vndk/v32/arm64/include/bionic/libc/kernel/uapi/linux/
Dbpf.h316 __aligned_u64 in_batch; member
/aosp_15_r20/prebuilts/vndk/v32/x86_64/include/bionic/libc/kernel/uapi/linux/
Dbpf.h316 __aligned_u64 in_batch; member
/aosp_15_r20/prebuilts/vndk/v32/arm/include/bionic/libc/kernel/uapi/linux/
Dbpf.h316 __aligned_u64 in_batch; member
/aosp_15_r20/prebuilts/vndk/v32/x86/include/bionic/libc/kernel/uapi/linux/
Dbpf.h316 __aligned_u64 in_batch; member
/aosp_15_r20/prebuilts/vndk/v31/x86/include/bionic/libc/kernel/uapi/linux/
Dbpf.h316 __aligned_u64 in_batch; member
/aosp_15_r20/prebuilts/vndk/v33/x86/include/bionic/libc/kernel/uapi/linux/
Dbpf.h329 __aligned_u64 in_batch; member
/aosp_15_r20/prebuilts/vndk/v33/arm/include/bionic/libc/kernel/uapi/linux/
Dbpf.h329 __aligned_u64 in_batch; member

123