Searched refs:MakeEdgePaddingConfig (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | resampler_ops.cc | 313 xla::MakeEdgePaddingConfig({{0, 0}, {1, 1}, {1, 1}, {0, 0}})); in CalculateGradData()
386 xla::MakeEdgePaddingConfig({{0, 0}, {1, 1}, {1, 1}, {0, 0}})); in CalculateGradWarp()
|
| H A D | fft_ops.cc | 91 xla::MakeEdgePaddingConfig(padding_sizes)); in Compile()
|
| H A D | stateful_random_ops.cc | 425 xla::MakeEdgePaddingConfig({{0, n}})); in PadRight()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | util.cc | 204 PaddingConfig MakeEdgePaddingConfig( in MakeEdgePaddingConfig() function
|
| H A D | util.h | 357 PaddingConfig MakeEdgePaddingConfig(
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | memory_space_assignment_test.cc | 2540 PaddingConfig padding_config = MakeEdgePaddingConfig({{0, 0}, {0, 1}}); in TEST_P()
|