Searched defs:tmp_pad (Results 1 – 1 of 1) sorted by relevance
73 Tensor tmp_pad = at::zeros({N, 1}, mask.options()); in NestedTensor_nested_tensor_from_mask() local104 Tensor tmp_pad = at::zeros({N, 1}, mask.options()); in NestedTensor_nested_tensor_from_mask_left_aligned() local