Home
last modified time | relevance | path

Searched refs:block_shape_height (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h3098 const int block_shape_height = block_shape_data[0]; in SpaceToBatchND() local
3113 if (out_h * block_shape_height + shift_h < padding_top || in SpaceToBatchND()
3114 out_h * block_shape_height + shift_h >= in SpaceToBatchND()
3124 (out_h * block_shape_height + shift_h) - padding_top, in SpaceToBatchND()
3157 const int block_shape_height = block_shape_data[0]; in BatchToSpaceND() local
3165 const int out_h = in_h * block_shape_height + in BatchToSpaceND()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5803 const int block_shape_height = block_shape_data[0]; in BatchToSpaceND() local
5815 block_shape_height, input_height, output_height, &in_h_start, in BatchToSpaceND()
5819 const int out_h = in_h * block_shape_height + in BatchToSpaceND()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h6131 const int block_shape_height = block_shape_data[0]; in BatchToSpaceND() local
6143 block_shape_height, input_height, output_height, &in_h_start, in BatchToSpaceND()
6147 const int out_h = in_h * block_shape_height + in BatchToSpaceND()