Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_4d_expander.cc76 int64_t output_spatial_dim = dim_nums.output_spatial_dimensions(i); in ExpandInstruction() local
Dhlo_cost_analysis.cc624 const int64_t output_spatial_dim = in GetConvolutionFlops() local
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
Ddynamic_padder.cc1132 int64_t output_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
Ddynamic_dimension_inference.cc757 int64_t output_spatial_dim = in HandleDynamicConvolutionForward() local