Home
last modified time | relevance | path

Searched defs:output_spatial_dim (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconvolution_4d_expander.cc76 int64_t output_spatial_dim = dim_nums.output_spatial_dimensions(i); in ExpandInstruction() local
H A Dspace_to_batch_converter.cc383 const int64_t output_spatial_dim = dim_numbers.output_spatial_dimensions( in IsForwardWindowDilatedConv() local
3654 const int64_t output_spatial_dim = dim_numbers.output_spatial_dimensions( in PerformSpaceToBatchOnConvolution() local
H A Dhlo_cost_analysis.cc624 const int64_t output_spatial_dim = in GetConvolutionFlops() local
H A Ddynamic_padder.cc1132 int64_t output_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
H A Ddynamic_dimension_inference.cc757 int64_t output_spatial_dim = in HandleDynamicConvolutionForward() local