Searched defs:is_pointwise (Results 1 – 3 of 3) sorted by relevance
86 const bool is_pointwise = (wei_shape[1] == wei_shape[2]) && wei_shape[1] == 1; in configure_G78_f32() local148 const bool is_pointwise = (wei_shape[1] == wei_shape[2]) && wei_shape[1] == 1; in configure_G78_f16() local282 const bool is_pointwise = (wei_shape[1] == wei_shape[2]) && wei_shape[1] == 1; in configure_G57_f32() local343 const bool is_pointwise = (wei_shape[1] == wei_shape[2]) && wei_shape[1] == 1; in configure_G57_f16() local
1223 mpy::object run_torch_function(Arena &A, mpy::handle orig, mpy::vector_args args, bool is_pointwise… in run_torch_function()1313 mpy::object __torch_function__(Arena &A, mpy::handle orig, mpy::vector_args args, bool is_pointwise… in __torch_function__()1369 bool is_pointwise = pointwise.contains(args[1]); in py___torch_function__() local2869 bool is_pointwise = false; member
46 inline bool is_pointwise(const IntArrayRef weight_size) { in is_pointwise() function