Home
last modified time | relevance | path

Searched refs:MakePadding (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc1219 /*pad_low=*/{xla::MakePadding({1}, {1}, {1}, Padding::kValid)[0].first},
1220 /*pad_high=*/{xla::MakePadding({1}, {1}, {1}, Padding::kValid)[0].second},
1225 /*pad_low=*/{xla::MakePadding({3}, {3}, {1}, Padding::kValid)[0].first},
1226 /*pad_high=*/{xla::MakePadding({3}, {3}, {1}, Padding::kValid)[0].second},
1231 /*pad_low=*/{xla::MakePadding({3}, {2}, {1}, Padding::kValid)[0].first},
1232 /*pad_high=*/{xla::MakePadding({3}, {2}, {1}, Padding::kValid)[0].second},
1237 /*pad_low=*/{xla::MakePadding({5}, {1}, {1}, Padding::kValid)[0].first},
1238 /*pad_high=*/{xla::MakePadding({5}, {1}, {1}, Padding::kValid)[0].second},
1243 /*pad_low=*/{xla::MakePadding({16}, {4}, {4}, Padding::kValid)[0].first},
1244 /*pad_high=*/{xla::MakePadding({16}, {4}, {4}, Padding::kValid)[0].second},
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.h59 std::vector<std::pair<int64, int64>> MakePadding(
Dpadding.cc41 std::vector<std::pair<int64, int64>> MakePadding( in MakePadding() function
Dpadding_test.cc30 return MakePadding({input_dimension}, {window_dimension}, {window_stride}, in ComputePadding()
Dxla_builder.cc1148 MakePadding(base_area_dimensions, window_dimensions, in ConvWithGeneralDimensions()
2007 MakePadding(AsInt64Slice(operand_shape.dimensions()), window_dimensions, in ReduceWindow()
2255 MakePadding(AsInt64Slice(operand_shape.dimensions()), window_dimensions, in SelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc199 xla::MakePadding(dim_lengths, window, stride, padding)); in ReduceWindow1DAdd()
250 xla::MakePadding(dim_lengths, window, stride, padding)); in ReduceWindow2DAdd()
257 auto padding_both = xla::MakePadding(dim_lengths, window, stride, padding); in ReduceWindow3DAdd()
307 xla::MakePadding(dim_lengths, window, stride, padding)); in ReduceWindow4DGeneric()
400 auto padding_both = xla::MakePadding(dim_lengths, window, stride, padding); in SelectAndScatter4DGePlus()
505 MakePadding(ordered_input_dimensions, ordered_kernel_dimensions, in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc188 return MakePadding(input_spatial_dimensions, kernel_size_spatial_dimensions, in MakeSpatialPadding()