Searched defs:input_spatial_dim (Results 1 – 5 of 5) sorted by relevance
75 int64_t input_spatial_dim = dim_nums.input_spatial_dimensions(i); in ExpandInstruction() local
976 int64_t input_spatial_dim = in RewriteDynamicConvolutionInputGrad() local1046 int64_t input_spatial_dim = in RewriteDynamicConvolutionForward() local1107 int64_t input_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
630 const int64_t input_spatial_dim = in GetConvolutionFlops() local
755 int64_t input_spatial_dim = in HandleDynamicConvolutionForward() local
476 auto input_spatial_dim = dnums.getInputSpatialDimensions(); in IsSamePadding() local