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
920 int64_t input_spatial_dim = in RewriteDynamicConvolutionInputGrad() local989 int64_t input_spatial_dim = in RewriteDynamicConvolutionForward() local1051 int64_t input_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
613 const int64_t input_spatial_dim = in HandleConvolution() local
726 int64_t input_spatial_dim = in HandleDynamicConvolutionForward() local
215 auto input_spatial_dim = dimension_numbers.input_spatial_dimensions().begin(); in HasCanonicalDimensionNumbers() local