| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | space_to_batch_nd.h | 64 const int input_batch_size = input1_shape.Dims(0); in SpaceToBatchND() local
80 int input_batch = out_b % input_batch_size; in SpaceToBatchND()
81 int shift_w = (out_b / input_batch_size) % block_shape_width; in SpaceToBatchND()
82 int shift_h = (out_b / input_batch_size) / block_shape_width; in SpaceToBatchND()
|
| D | batch_to_space_nd.h | 65 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local
73 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_builder.cc | 102 int input_batch_size, int input_height_size, int input_width_size, in PopulateSpaceToBatchOutputDims() argument
108 input_batch_size * dilation_factors_h_w[0] * dilation_factors_h_w[1]); in PopulateSpaceToBatchOutputDims()
141 int input_batch_size, input_height_size, input_width_size, input_depth_size; in BuildDilatedDwConv() local
142 GetDims(&input_batch_size, &input_height_size, &input_width_size, in BuildDilatedDwConv()
170 input_batch_size, input_height_size, input_width_size, input_depth_size, in BuildDilatedDwConv()
247 int input_batch_size, input_height_size, input_width_size, input_depth_size; in BuildSplittedDwConv() local
248 GetDims(&input_batch_size, &input_height_size, &input_width_size, in BuildSplittedDwConv()
270 {input_batch_size, input_height_size, input_width_size, in BuildSplittedDwConv()
377 int input_batch_size, input_height_size, input_width_size, input_depth_size; in PopulateSubGraph() local
378 GetDims(&input_batch_size, &input_height_size, &input_width_size, in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batchtospace_op.cc | 139 int64_t input_batch_size = orig_input_batch_size; in BatchToSpaceOpCompute() local
142 input_batch_size *= size; in BatchToSpaceOpCompute()
145 internal_input_shape.AddDim(input_batch_size); in BatchToSpaceOpCompute()
146 internal_output_shape.AddDim(input_batch_size / block_shape_product); in BatchToSpaceOpCompute()
|
| D | spacetobatch_op.cc | 149 int64_t input_batch_size = orig_input_tensor.dim_size(0); in SpaceToBatchOpCompute() local
152 input_batch_size *= size; in SpaceToBatchOpCompute()
155 internal_input_shape.AddDim(input_batch_size); in SpaceToBatchOpCompute()
156 internal_output_shape.AddDim(input_batch_size * block_shape_product); in SpaceToBatchOpCompute()
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 379 const int32_t input_batch_size = input_shape[0]; in SpaceToBatchNDRank4() local
393 const int32_t output_batch_size = input_batch_size * block_size; in SpaceToBatchNDRank4()
402 const int32_t input_batch = out_b % input_batch_size; in SpaceToBatchNDRank4()
403 const int32_t shift_w = (out_b / input_batch_size) % block_shape_width; in SpaceToBatchNDRank4()
404 const int32_t shift_h = (out_b / input_batch_size) / block_shape_width; in SpaceToBatchNDRank4()
450 const int32_t input_batch_size = input_shape[0]; in BatchToSpaceNDRank4() local
463 input_batch_size / (block_shape_width * block_shape_height); in BatchToSpaceNDRank4()
469 for (int32_t in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceNDRank4()
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 379 const int32_t input_batch_size = input_shape[0]; in SpaceToBatchNDRank4() local
393 const int32_t output_batch_size = input_batch_size * block_size; in SpaceToBatchNDRank4()
402 const int32_t input_batch = out_b % input_batch_size; in SpaceToBatchNDRank4()
403 const int32_t shift_w = (out_b / input_batch_size) % block_shape_width; in SpaceToBatchNDRank4()
404 const int32_t shift_h = (out_b / input_batch_size) / block_shape_width; in SpaceToBatchNDRank4()
450 const int32_t input_batch_size = input_shape[0]; in BatchToSpaceNDRank4() local
463 input_batch_size / (block_shape_width * block_shape_height); in BatchToSpaceNDRank4()
469 for (int32_t in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceNDRank4()
|
| D | onedevice_train_palm_detection_filter_inference.cc | 380 const int32_t input_batch_size = input_shape[0]; in SpaceToBatchNDRank4() local
394 const int32_t output_batch_size = input_batch_size * block_size; in SpaceToBatchNDRank4()
403 const int32_t input_batch = out_b % input_batch_size; in SpaceToBatchNDRank4()
404 const int32_t shift_w = (out_b / input_batch_size) % block_shape_width; in SpaceToBatchNDRank4()
405 const int32_t shift_h = (out_b / input_batch_size) / block_shape_width; in SpaceToBatchNDRank4()
451 const int32_t input_batch_size = input_shape[0]; in BatchToSpaceNDRank4() local
464 input_batch_size / (block_shape_width * block_shape_height); in BatchToSpaceNDRank4()
470 for (int32_t in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceNDRank4()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 57 const int64_t input_batch_size = in PartitionConvolutionWithBatchGroupCount() local
61 if (input_batch_size != kernel_output_feature_size || in PartitionConvolutionWithBatchGroupCount()
62 original_hlo->batch_group_count() != input_batch_size) { in PartitionConvolutionWithBatchGroupCount()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | depthwise_conv_hybrid_3x3_filter.h | 3095 const int64_t input_batch_size = params.input_row_size * params.input_height;
3148 const int8* input_ptr = input_data + b * input_batch_size;
|
| D | depthwise_conv_3x3_filter.h | 2934 const int64_t input_batch_size = params.input_row_size * params.input_height;
2990 const int8* input_ptr = input_data + b * input_batch_size;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 960 const int64_t input_batch_size =
963 const int64_t batch_group_size = input_batch_size / batch_group_count;
982 batch_group_count > 1 ? output_z_size / input_batch_size : 1;
1056 ((batch_group_index * batch_group_size) % input_batch_size) *
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | depthwiseconv_uint8_3x3_filter.h | 5645 const int64_t input_batch_size = params.input_row_size * params.input_height;
5700 const uint8* input_ptr = input_data + b * input_batch_size;
|
| D | optimized_ops.h | 4230 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local
4239 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND()
|