Searched defs:pad_dim (Results 1 – 7 of 7) sorted by relevance
236 auto pad_dim = [](Shape* s, int64_t dim) { in TryResolvePaddedShapesForTensorCore() local358 auto pad_dim = [&](Shape* s, int64_t dim, int64_t cur_vect_size) { in TryResolvePaddedShapesForIntegerConvolution() local
42 auto pad_dim = [&](Shape& s, int dim) { in PadForGemm() local
17 const int pad_dim = padding.size(); in pad2d() local
415 def pad_dim(x: Tensor, padded_length: int, dim: int) -> Tensor: function
131 def pad_dim(x: Tensor, padded_length: int, dim: int) -> Tensor: function
2312 const auto& pad_dim = hlo->padding_config().dimensions(i); in EmitElementalPad() local
5914 const auto& pad_dim = pad_config.dimensions(i); in HandleReduceWindow() local