Searched defs:input_batch_size (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | conv_2d_builder.cc | 102 int input_batch_size, int input_height_size, int input_width_size, in PopulateSpaceToBatchOutputDims()
141 int input_batch_size, input_height_size, input_width_size, input_depth_size; in BuildDilatedDwConv() local
247 int input_batch_size, input_height_size, input_width_size, input_depth_size; in BuildSplittedDwConv() local
377 int input_batch_size, input_height_size, input_width_size, input_depth_size; in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | batch_to_space_nd.h | 65 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local
|
| H A D | space_to_batch_nd.h | 64 const int input_batch_size = input1_shape.Dims(0); in SpaceToBatchND() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | spacetobatch_op.cc | 149 int64_t input_batch_size = orig_input_tensor.dim_size(0); in SpaceToBatchOpCompute() local
|
| H A D | batchtospace_op.cc | 139 int64_t input_batch_size = orig_input_batch_size; in BatchToSpaceOpCompute() local
|
| /aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| H A D | onedevice_train_palm_detection_filter_inference_beta.cc | 379 const int32_t input_batch_size = input_shape[0]; in SpaceToBatchNDRank4() local
450 const int32_t input_batch_size = input_shape[0]; in BatchToSpaceNDRank4() local
|
| H A D | onedevice_train_palm_detection_filter_inference.cc | 380 const int32_t input_batch_size = input_shape[0]; in SpaceToBatchNDRank4() local
451 const int32_t input_batch_size = input_shape[0]; in BatchToSpaceNDRank4() local
|
| H A D | onedevice_train_palm_detection_filter_inference_v2.cc | 379 const int32_t input_batch_size = input_shape[0]; in SpaceToBatchNDRank4() local
450 const int32_t input_batch_size = input_shape[0]; in BatchToSpaceNDRank4() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | convolution_handler.cc | 57 const int64_t input_batch_size = in PartitionConvolutionWithBatchGroupCount() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | optimized_ops.h | 4230 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local
|