Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc471 const int64 output_spatial_dim = in HandleConvolution() local
473 const int64 output_limit = result_shape.dimensions(output_spatial_dim); in HandleConvolution()
Dhlo_evaluator_typed_visitor.h1119 const int64 output_spatial_dim = dnums.output_spatial_dimensions(ki); in HandleConvolution() local
1125 out_index[output_spatial_dim] * window_dim.stride() - in HandleConvolution()