Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc499 const int64 undilated_index = output_idx * window_dim.stride() - in HandleConvolution() local
507 ? undilated_index / window_dim.base_dilation() in HandleConvolution()
508 : undilated_index; in HandleConvolution()
511 if (undilated_index != lhs_spatial_index * window_dim.base_dilation()) { in HandleConvolution()
Dhlo_evaluator_typed_visitor.h1124 const int64 undilated_index = in HandleConvolution() local
1131 undilated_index % window_dim.base_dilation() != 0) { in HandleConvolution()
1139 lhs_spatial_index = undilated_index / window_dim.base_dilation(); in HandleConvolution()
1141 lhs_spatial_index = undilated_index; in HandleConvolution()