Home
last modified time | relevance | path

Searched refs:low_pad (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc420 const int64_t low_pad = convolution->window() in IsConvSuitableForSpaceToBatch() local
433 if (low_pad == 0) { in IsConvSuitableForSpaceToBatch()
437 } else if (low_pad != c.base_dilation_factor - 1 && in IsConvSuitableForSpaceToBatch()
438 low_pad != c.base_dilation_factor) { in IsConvSuitableForSpaceToBatch()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc4771 const int64_t low_pad = lhs->shape().dimensions(dim); in PreprocessHlos() local
4773 padding_config_dim->set_edge_padding_low(low_pad); in PreprocessHlos()
4793 padded_size - (pad_pattern->lhs_slice_start + low_pad); in PreprocessHlos()
4826 padded_size - (pad_pattern->rhs_slice_start + low_pad + high_pad); in PreprocessHlos()
4849 {low_pad, rotate_lhs, Comparison::Direction::kLt}, in PreprocessHlos()