Home
last modified time | relevance | path

Searched defs:output_image_size (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dtranspose_conv.h42 const int output_image_size = output_height * output_width; in TransposeConvV2() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_grad_op.cc423 const int64 output_image_size = in operator ()() local
801 const int64 output_image_size = in ComputeBackpropFilter() local
912 const int64 output_image_size = in operator ()() local
Dpooling_ops_common.h249 const int32 output_image_size = out_height * out_width * params.depth; in SpatialMaxPool() local
Ddepthwise_conv_op.cc197 const int64 output_image_size = in operator ()() local
Dconv_grad_ops_3d.cc391 const int64 output_image_size = dims.spatial_dims[0].output_size * in Compute() local
858 const int64 output_image_size = dims.spatial_dims[0].output_size * in Compute() local
Dmaxpooling_op.cc125 const int32 output_image_size = out_height * out_width * depth; in SpatialMaxPoolWithArgMaxHelper() local
605 const int32 output_image_size = out_height * out_width * params.depth; in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc593 const int32 output_image_size = in launch() local
Dconv_grad_filter_ops.cc499 const int output_image_size = in Compute() local
Ddeep_conv2d.cc1115 const int64 output_image_size = args.out_rows * args.out_cols * out_depth; in operator ()() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5729 const int output_image_size = output_height * output_width; in TransposeConvV2() local
5955 const int output_image_size = output_height * output_width; in TransposeConvV2() local