Searched refs:pad_dim (Results 1 – 3 of 3) sorted by relevance
174 auto pad_dim = [](Shape* s, int64 dim) { in PadForTensorCores() local177 pad_dim(new_input_shape, dnums.input_feature_dimension()); in PadForTensorCores()178 pad_dim(new_filter_shape, dnums.kernel_input_feature_dimension()); in PadForTensorCores()179 pad_dim(new_filter_shape, dnums.kernel_output_feature_dimension()); in PadForTensorCores()180 pad_dim(new_output_shape, dnums.output_feature_dimension()); in PadForTensorCores()
2060 const auto& pad_dim = hlo->padding_config().dimensions(i); in EmitElementalPad() local2062 Sub(multi_index[i], index_typed_const(pad_dim.edge_padding_low())); in EmitElementalPad()2069 index_typed_const(pad_dim.interior_padding() + 1))), in EmitElementalPad()2072 SDiv(multi_index[i], index_typed_const(pad_dim.interior_padding() + 1)); in EmitElementalPad()
3291 const auto& pad_dim = pad_config.dimensions(i); in HandleReduceWindow() local3294 pad_dim.edge_padding_low()); in HandleReduceWindow()3296 pad_dim.edge_padding_high()); in HandleReduceWindow()