Home
last modified time | relevance | path

Searched refs:in_batch (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbatch_to_space_nd.h73 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND() local
74 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND()
75 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND()
91 input1_data + Offset(input1_shape, in_batch, in_h, in_w, 0); in BatchToSpaceND()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc44 const int64_t in_batch = GetTensorDim(tensor_in, data_format, 'N'); in Compute() local
74 input_desc.set_count(in_batch) in Compute()
78 output_desc.set_count(in_batch) in Compute()
145 const int64_t in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute() local
208 orig_output_desc.set_count(in_batch) in Compute()
213 orig_input_desc.set_count(in_batch) in Compute()
Dconv_ops_3d.cc147 const int64_t in_batch = GetTensorDim(input, data_format_, 'N'); in Compute() local
183 data_format_, in_batch, {{out[0], out[1], out[2]}}, out_depth); in Compute()
236 const int64_t in_batch = GetTensorDim(input, data_format, 'N'); in launch() local
270 const uint64 m = in_batch * in_planes * in_rows * in_cols; in launch()
291 const uint64 m = in_batch; in launch()
325 data_format, in_batch, {{new_in_planes, new_in_rows, new_in_cols}}, in launch()
360 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth); in launch()
400 input_desc.set_count(in_batch) in launch()
407 output_desc.set_count(in_batch) in launch()
462 ShapeFromFormat(FORMAT_NCHW, in_batch, in launch()
[all …]
Dfused_batch_norm_op.cc131 const int64_t in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
137 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
142 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
278 const int64_t in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
284 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
289 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
373 const int64_t in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()() local
379 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
384 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
389 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
[all …]
Dfractional_avg_pool_op.cc270 const int64_t in_batch = orig_input_tensor_shape_flat(0); in Compute() local
275 context, in_batch != 0, in Compute()
303 in_cols * in_rows * in_batch); in Compute()
Dconv_ops_fused_impl.h467 const int64_t in_batch = GetTensorDim(input, params.data_format, 'N');
515 ShapeFromFormat(params.data_format, in_batch,
550 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths);
580 input_desc.set_count(in_batch)
653 in_batch, // batch
Dconv_ops.cc796 const int64_t in_batch = GetTensorDim(input, data_format, 'N'); in operator ()() local
818 const uint64 m = in_batch * in_rows * in_cols; in operator ()()
839 const uint64 m = in_batch; in operator ()()
930 ShapeFromFormat(data_format, in_batch, new_in_rows, in operator ()()
963 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths); in operator ()()
1002 input_desc.set_count(in_batch) in operator ()()
1089 ConvParameters conv_parameters = {in_batch, // batch in operator ()()
Dpooling_ops_3d.cc167 const int64_t in_batch = GetTensorDim(tensor_in, data_format_, 'N'); in Compute() local
185 TensorShape out_shape = ShapeFromFormat(data_format_, in_batch, in Compute()
/external/ComputeLibrary/src/core/NEON/kernels/convolution/common/
Dshims.hpp174 const T* const in_batch = in + n*in_batch_stride; in nchw_to_nhwc() local
179 const T* const in_row = in_batch + i*in_row_stride; in nchw_to_nhwc()
345 const T* const in_batch = in + n*in_batch_stride; in nchw_to_nhwc() local
350 const T* const in_row = in_batch + i*in_row_stride; in nchw_to_nhwc()
405 const T* const in_batch = in + n*in_batch_stride; in nhwc_to_nchw() local
411 const T* const in_i = in_batch + i*in_row_stride; in nhwc_to_nchw()
588 const T* const in_batch = in + n*in_batch_stride; in nhwc_to_nchw() local
594 const T* const in_i = in_batch + i*in_row_stride; in nhwc_to_nchw()
/external/libbpf/src/
Dbpf.c667 static int bpf_map_batch_common(int cmd, int fd, void *in_batch, in bpf_map_batch_common() argument
680 attr.batch.in_batch = ptr_to_u64(in_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
705 return bpf_map_batch_common(BPF_MAP_LOOKUP_BATCH, fd, in_batch, 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,
332 LIBBPF_API int bpf_map_lookup_and_delete_batch(int fd, void *in_batch,
/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/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
Donedevice_train_palm_detection_filter_inference.cc470 for (int32_t in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceNDRank4() local
471 const int32_t out_batch = in_batch % output_batch_size; in BatchToSpaceNDRank4()
472 const int32_t spatial_offset = in_batch / output_batch_size; in BatchToSpaceNDRank4()
491 (((in_batch * input_height + in_h) * input_width + in_w) * in BatchToSpaceNDRank4()
Donedevice_train_palm_detection_filter_inference_beta.cc469 for (int32_t in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceNDRank4() local
470 const int32_t out_batch = in_batch % output_batch_size; in BatchToSpaceNDRank4()
471 const int32_t spatial_offset = in_batch / output_batch_size; in BatchToSpaceNDRank4()
490 (((in_batch * input_height + in_h) * input_width + in_w) * in BatchToSpaceNDRank4()
Donedevice_train_palm_detection_filter_inference_v2.cc469 for (int32_t in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceNDRank4() local
470 const int32_t out_batch = in_batch % output_batch_size; in BatchToSpaceNDRank4()
471 const int32_t spatial_offset = in_batch / output_batch_size; in BatchToSpaceNDRank4()
490 (((in_batch * input_height + in_h) * input_width + in_w) * in BatchToSpaceNDRank4()
/external/bpftool/include/uapi/linux/
Dbpf.h1305 __aligned_u64 in_batch; /* start batch, member
/external/kernel-headers/original/uapi/linux/
Dbpf.h1298 __aligned_u64 in_batch; /* start batch, member
/external/libbpf/include/uapi/linux/
Dbpf.h1305 __aligned_u64 in_batch; /* start batch, member
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h4562 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND() local
4563 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND()
4564 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND()
4593 input1_data + Offset(input1_shape, in_batch, in_h, in_w, 0); in BatchToSpaceND()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc4071 int in_batch = 10; in TEST_F() member
4132 in_dims.push_back(options.in_batch); in TEST_F()
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_510.h10390 __u64 in_batch; member
Dvmlinux.h10390 __u64 in_batch; member
/external/libbpf/.github/actions/build-selftests/
Dvmlinux.h25381 __u64 in_batch; member

12