Searched defs:stridemat (Results 1 – 4 of 4) sorted by relevance
487 & stridemat = input_ptr->get_nested_strides(); in native_dropout_nested() local598 & stridemat = self_ptr->get_nested_strides(); in transpose_nested() local627 const Tensor& stridemat = self_ptr->get_nested_strides(); in squeeze_dim_nested() local674 const Tensor& stridemat = self_ptr->get_nested_strides(); in unsqueeze_nested() local
141 stridemat = self_ptr->get_nested_strides(); in clone_nested() local
148 const Tensor& stridemat = self_ptr->get_nested_strides(); in NestedTensor_get_strides() local
230 &stridemat = nt->get_nested_strides(); in nested_tensor_impl_is_contiguous() local