Home
last modified time | relevance | path

Searched defs:low_padding (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc367 constexpr int64_t low_padding = 0; in XLA_TEST_P() local
394 int64 low_padding[2] = {-1, -2}; in XLA_TEST_P() local
422 int64 low_padding[2] = {4, -1}; in XLA_TEST_P() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc408 int64_t activations_batch_dim, int64_t old_batch_size, int64_t low_padding, in HaloDuplicateWithSlice()
2424 int64_t activations_batch_dim, int64_t high_padding, int64_t low_padding, in SplitSpaceHelper()
2480 int64_t high_padding, int64_t low_padding, in SplitSpace()
Dalgebraic_simplifier_test.cc3315 int64 low_padding[2] = {-1, -2}; in TEST_F() local
3363 int64 low_padding[2] = {-1, -2}; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1131 int64_t low_padding = 0; in HaloExchangeToPadOnLeft() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3025 DimVector low_padding(num_dims, 0); in matchAndRewrite() local
6575 llvm::SmallVector<int64_t, 4> low_padding(rank, 0); in matchAndRewrite() local