Searched refs:multi_tensor_apply (Results 1 – 11 of 11) sorted by relevance
49 multi_tensor_apply<4>( in foreach_tensor_lerp_ternary_cuda()80 multi_tensor_apply<3>( in foreach_tensor_lerp_ternary_cuda_()116 multi_tensor_apply<3>( in foreach_tensor_lerp_list_cuda()148 multi_tensor_apply<2>( in foreach_tensor_lerp_list_cuda_()
49 multi_tensor_apply<4>( in foreach_pointwise_op()81 multi_tensor_apply<3>( in foreach_pointwise_op_()113 multi_tensor_apply<3, opmath_t>( in foreach_pointwise_op_()152 multi_tensor_apply<4, opmath_t>( in foreach_pointwise_op()
175 multi_tensor_apply<3>( in _fused_sgd_with_momentum_kernel_cuda_()249 multi_tensor_apply<3>( in _fused_sgd_with_momentum_kernel_cuda_()315 multi_tensor_apply<2>( in _fused_sgd_kernel_cuda_()407 multi_tensor_apply<2>( in _fused_sgd_kernel_cuda_()
44 multi_tensor_apply<3>( in foreach_tensor_list_op()67 multi_tensor_apply<2>( in foreach_tensor_list_op_()423 multi_tensor_apply<2>( in foreach_tensor_copy_list_kernel_cuda_()438 multi_tensor_apply<2>( in foreach_tensor_copy_list_kernel_cuda_()
59 multi_tensor_apply<2>( in foreach_unary_op()76 multi_tensor_apply<1>( in foreach_unary_op_()398 multi_tensor_apply<1>( in foreach_tensor_zero_cuda_()
49 multi_tensor_apply<2>( in foreach_binary_op()84 multi_tensor_apply<1>( in foreach_binary_op_()
39 multi_tensor_apply<2, opmath_t>( in foreach_binary_op()58 multi_tensor_apply<1, opmath_t>( in foreach_binary_op_()
39 multi_tensor_apply<2>( in foreach_binary_op()57 multi_tensor_apply<1>( in foreach_binary_op_()
181 multi_tensor_apply<1>( in foreach_tensor_max_cuda()479 multi_tensor_apply<1>( in foreach_tensor_norm_cuda()485 multi_tensor_apply<1>( in foreach_tensor_norm_cuda()491 multi_tensor_apply<1>( in foreach_tensor_norm_cuda()
126 void multi_tensor_apply( in multi_tensor_apply() function218 void multi_tensor_apply( in multi_tensor_apply() function
160 multi_tensor_apply<1>(tensor_lists, in _amp_foreach_non_finite_check_and_unscale_cuda_()