Searched defs:low_padding (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | pad_test.cc | 365 constexpr int64_t low_padding = 0; in XLA_TEST_P() local
392 int64_t low_padding[2] = {-1, -2}; in XLA_TEST_P() local
420 int64_t low_padding[2] = {4, -1}; in XLA_TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | space_to_batch_converter.cc | 512 int64_t activations_batch_dim, int64_t low_padding, int64_t halo_size, in HaloDuplicateWithSlice()
2903 int64_t activations_batch_dim, int64_t high_padding, int64_t low_padding, in PadAndSplitSpace()
2937 int64_t& activations_batch_dim, int64_t high_padding, int64_t low_padding, in SplitSpace()
|
| H A D | algebraic_simplifier_test.cc | 3690 int64_t low_padding[2] = {-1, -2}; in TEST_F() local
3766 int64_t low_padding[2] = {-1, -2}; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner_util.cc | 1047 int64_t low_padding = 0; in HaloExchangeToPadOnLeft() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 2809 DimVector low_padding(num_dims, 0); in matchAndRewrite() local
6342 llvm::SmallVector<int64_t, 4> low_padding(rank, 0); in matchAndRewrite() local
|