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.h47 const int output_image_size = output_height * output_width; in TransposeConvV2() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_grad_op.cc427 const int64_t output_image_size = in operator ()() local
805 const int64_t output_image_size = in ComputeBackpropFilter() local
916 const int64_t output_image_size = in operator ()() local
Dpooling_ops_common.h247 const int32_t output_image_size = in SpatialMaxPool() local
Ddepthwise_conv_op.cc200 const int64_t output_image_size = in operator ()() local
Dmaxpooling_op.cc127 const int32_t output_image_size = out_height * out_width * depth; in SpatialMaxPoolWithArgMaxHelper() local
629 const int32_t output_image_size = out_height * out_width * params.depth; in SpatialMaxPoolGradGrad() local
Dconv_grad_ops_3d.cc439 const int64_t output_image_size = dims.spatial_dims[0].output_size * in Compute() local
963 const int64_t output_image_size = dims.spatial_dims[0].output_size * in Compute() local
Dpooling_ops_3d.cc616 const int32_t output_image_size = in launch() local
Dconv_grad_filter_ops.cc511 const int output_image_size = in Compute() local
Ddeep_conv2d.cc1133 const int64_t output_image_size = in operator ()() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h4968 const int output_image_size = output_height * output_width; in TransposeConvV2() local
5467 const int output_image_size = output_height * output_width; in TransposeConvV2() local