Home
last modified time | relevance | path

Searched refs:out_batch (Results 1 – 23 of 23) sorted by relevance

/external/ComputeLibrary/src/core/NEON/kernels/convolution/common/
Dshims.hpp175 T* const out_batch = out + n*out_batch_stride; in nchw_to_nhwc() local
180 T* const out_row = out_batch + i*out_row_stride; in nchw_to_nhwc()
346 T* const out_batch = out + n*out_batch_stride; in nchw_to_nhwc() local
351 T* const out_row = out_batch + i*out_row_stride; in nchw_to_nhwc()
406 T* const out_batch = out + n*out_batch_stride; in nhwc_to_nchw() local
412 T* const out_i = out_batch + i*out_row_stride; in nhwc_to_nchw()
589 T* const out_batch = out + n*out_batch_stride; in nhwc_to_nchw() local
595 T* const out_i = out_batch + i*out_row_stride; in nhwc_to_nchw()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbatch_to_space_nd.h74 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local
89 T* out = output_data + Offset(output_shape, out_batch, out_h, out_w, 0); in BatchToSpaceND()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h391 int out_batch = GetTensorDim(input_shape, data_format_, 'N'); variable
471 ? ShapeFromFormat(data_format_, out_batch, out_rows, out_cols,
473 : ShapeFromFormat(data_format_, out_batch,
491 mkldnn_sizes[MklDnnDims::Dim_N] = out_batch;
498 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = out_batch;
/external/tensorflow/tensorflow/core/kernels/
Dfractional_avg_pool_op.cc248 const int64_t out_batch = out_backprop.dim_size(0); in Compute() local
306 out_cols * out_rows * out_batch); in Compute()
311 for (int64_t b = 0; b < out_batch; ++b) { in Compute()
Dconv_ops_fused_impl.h476 const int64_t out_batch = GetTensorDim(*output, params.data_format, 'N');
605 output_desc.set_count(out_batch)
629 ShapeFromFormat(FORMAT_NCHW, out_batch, out_rows,
Dconv_ops.cc880 const int64_t out_batch = GetTensorDim(*output, data_format, 'N'); in operator ()() local
1008 output_desc.set_count(out_batch) in operator ()()
1066 ShapeFromFormat(compute_data_format, out_batch, in operator ()()
/external/libbpf/src/
Dbpf.c668 void *out_batch, void *keys, void *values, in bpf_map_batch_common() argument
681 attr.batch.out_batch = ptr_to_u64(out_batch); in bpf_map_batch_common()
701 int bpf_map_lookup_batch(int fd, void *in_batch, void *out_batch, void *keys, in bpf_map_lookup_batch() argument
706 out_batch, keys, values, count, opts); in bpf_map_lookup_batch()
709 int bpf_map_lookup_and_delete_batch(int fd, void *in_batch, void *out_batch, in bpf_map_lookup_and_delete_batch() argument
714 fd, in_batch, out_batch, keys, values, in bpf_map_lookup_and_delete_batch()
Dbpf.h306 LIBBPF_API int bpf_map_lookup_batch(int fd, void *in_batch, void *out_batch,
333 void *out_batch, void *keys,
/external/bcc/src/cc/
Dlibbpf.h164 int bpf_lookup_batch(int fd, __u32 *in_batch, __u32 *out_batch, void *keys,
168 int bpf_lookup_and_delete_batch(int fd, __u32 *in_batch, __u32 *out_batch,
Dlibbpf.c415 int bpf_lookup_batch(int fd, __u32 *in_batch, __u32 *out_batch, void *keys, in bpf_lookup_batch() argument
418 return bpf_map_lookup_batch(fd, in_batch, out_batch, keys, values, count, in bpf_lookup_batch()
432 int bpf_lookup_and_delete_batch(int fd, __u32 *in_batch, __u32 *out_batch, in bpf_lookup_and_delete_batch() argument
435 return bpf_map_lookup_and_delete_batch(fd, in_batch, out_batch, keys, values, in bpf_lookup_and_delete_batch()
/external/bcc/src/cc/includes/
Dlibbpf.h164 int bpf_lookup_batch(int fd, __u32 *in_batch, __u32 *out_batch, void *keys,
168 int bpf_lookup_and_delete_batch(int fd, __u32 *in_batch, __u32 *out_batch,
/external/bpftool/include/uapi/linux/
Dbpf.h1308 __aligned_u64 out_batch; /* output: next start batch */ member
/external/kernel-headers/original/uapi/linux/
Dbpf.h1301 __aligned_u64 out_batch; /* output: next start batch */ member
/external/libbpf/include/uapi/linux/
Dbpf.h1308 __aligned_u64 out_batch; /* output: next start batch */ member
/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
Donedevice_train_palm_detection_filter_inference.cc471 const int32_t out_batch = in_batch % output_batch_size; in BatchToSpaceNDRank4() local
487 (((out_batch * output_height + out_h) * output_width + out_w) * in BatchToSpaceNDRank4()
Donedevice_train_palm_detection_filter_inference_beta.cc470 const int32_t out_batch = in_batch % output_batch_size; in BatchToSpaceNDRank4() local
486 (((out_batch * output_height + out_h) * output_width + out_w) * in BatchToSpaceNDRank4()
Donedevice_train_palm_detection_filter_inference_v2.cc470 const int32_t out_batch = in_batch % output_batch_size; in BatchToSpaceNDRank4() local
486 (((out_batch * output_height + out_h) * output_width + out_w) * in BatchToSpaceNDRank4()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h4563 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local
4591 T* out = output_data + Offset(output_shape, out_batch, out_h, out_w, 0); in BatchToSpaceND()
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_510.h10391 __u64 out_batch; member
Dvmlinux.h10391 __u64 out_batch; member
/external/libbpf/.github/actions/build-selftests/
Dvmlinux.h25382 __u64 out_batch; member
/external/bcc/libbpf-tools/arm64/
Dvmlinux.h9366 __u64 out_batch; member
Dvmlinux_510.h9366 __u64 out_batch; member