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 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.cc345 int64 activations_batch_dim, int64 old_batch_size, int64 low_padding, in HaloDuplicateWithSlice()
2039 int64 activations_batch_dim, int64 high_padding, int64 low_padding, in SplitSpaceHelper()
2095 int64 low_padding, int64 spatial_split_size, in SplitSpace()
Dalgebraic_simplifier_test.cc3282 int64 low_padding[2] = {-1, -2}; in TEST_F() local
3330 int64 low_padding[2] = {-1, -2}; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1085 int64 low_padding = 0; in HaloExchangeToPadOnLeft() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2411 DimVector low_padding(num_dims, 0); in matchAndRewrite() local
5434 llvm::SmallVector<int64_t, 4> low_padding(rank, 0); in matchAndRewrite() local