Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc1298 /*pad_low=*/{xla::MakePadding({1}, {1}, {1}, Padding::kValid)[0].first},
1299 /*pad_high=*/{xla::MakePadding({1}, {1}, {1}, Padding::kValid)[0].second},
1304 /*pad_low=*/{xla::MakePadding({3}, {3}, {1}, Padding::kValid)[0].first},
1305 /*pad_high=*/{xla::MakePadding({3}, {3}, {1}, Padding::kValid)[0].second},
1310 /*pad_low=*/{xla::MakePadding({3}, {2}, {1}, Padding::kValid)[0].first},
1311 /*pad_high=*/{xla::MakePadding({3}, {2}, {1}, Padding::kValid)[0].second},
1316 /*pad_low=*/{xla::MakePadding({5}, {1}, {1}, Padding::kValid)[0].first},
1317 /*pad_high=*/{xla::MakePadding({5}, {1}, {1}, Padding::kValid)[0].second},
1322 /*pad_low=*/{xla::MakePadding({16}, {4}, {4}, Padding::kValid)[0].first},
1323 /*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.cc1427 MakePadding(base_area_dimensions, window_dimensions, in ConvWithGeneralDimensions()
2513 MakePadding(AsInt64Slice(operand_shape->dimensions()), in ReduceWindow()
3036 MakePadding(AsInt64Slice(operand_shape->dimensions()), in SelectAndScatter()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dmake_padding_test.cc35 TEST(MakePadding, Smoke) { in TEST() argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dtree_reduction_rewriter.cc72 MakePadding(AsInt64Slice(input_shape.dimensions()), window_dimensions, in HandleReduce()
Dshape_inference_test.cc1112 MakePadding(AsInt64Slice(f32_arg_shape.dimensions()), window_dimensions, in TEST_F()
1174 MakePadding(AsInt64Slice(f32_arg_shape.dimensions()), window_dimensions, in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc184 xla::MakePadding(dim_lengths, window, stride, padding)); in ReduceWindow1DAdd()
191 auto padding_both = xla::MakePadding(dim_lengths, window, stride, padding); in ReduceWindow3DAdd()
241 xla::MakePadding(dim_lengths, window, stride, padding)); in ReduceWindow4DGeneric()
334 auto padding_both = xla::MakePadding(dim_lengths, window, stride, padding); in SelectAndScatter4DGePlus()
439 MakePadding(ordered_input_dimensions, ordered_kernel_dimensions, in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc189 return MakePadding(input_spatial_dimensions, kernel_size_spatial_dimensions, in MakeSpatialPadding()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2157 PaddingArray paddings = ::xla::MakePadding(input_shape, window_shape, strides, in GetReduceWindowPaddingAsArray()