Home
last modified time | relevance | path

Searched refs:input_spatial_size (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc962 int64_t input_spatial_size = in MatchResizeOp() local
968 tensorflow::MathUtil::GCD(static_cast<uint64_t>(input_spatial_size), in MatchResizeOp()
970 if ((input_spatial_size % gcd != 0) || in MatchResizeOp()
971 (input_spatial_size / gcd != stride) || (dilation - 1 != padding)) { in MatchResizeOp()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h8231 const int input_spatial_size = in Conv3DTranspose() local
8239 const int input_offset = input_spatial_size * input_channel; in Conv3DTranspose()
8271 rhs_params.cols = input_spatial_size; in Conv3DTranspose()
8275 dst_params.cols = input_spatial_size; in Conv3DTranspose()