Searched defs:bias_bdim (Results 1 – 2 of 2) sorted by relevance
49 const std::optional<Tensor>& bias_opt, std::optional<int64_t> bias_bdim, in batch_norm_batch_rule()475 const Tensor& bias, std::optional<int64_t> bias_bdim) { in _check_layer_norm_inputs()492 const std::optional<Tensor>& bias_opt, std::optional<int64_t> bias_bdim, in native_layer_norm_batch_rule()604 std::optional<int64_t> bias_bdim; in native_layer_norm_backward_plumbing() local
20 …ional<int64_t> rhs_bdim, const std::optional<Tensor>& bias, std::optional<int64_t> bias_bdim, c10:… in convolution_batch_rule()