Searched defs:mb_weight (Results 1 – 1 of 1) sorted by relevance
57 std::optional<Tensor> mb_weight, in q_batch_norm1d_impl()165 std::optional<Tensor> mb_weight, in q_batch_norm2d_impl()259 std::optional<Tensor> mb_weight, in q_batch_norm3d_impl()356 std::optional<Tensor> mb_weight, in q_batch_norm_impl()