Searched refs:weight_dim (Results 1 – 3 of 3) sorted by relevance
53 py::tuple weight_dim = py::make_tuple(64, 3, 3, 3); in ConstructOpExecInfo() local54 py::object weight = tensor_py_module(np_float32(np_ones(weight_dim))); in ConstructOpExecInfo()
41 size_t weight_dim = weight_shape.size(); in InitKernel() local42 if (src_dim < kShapeSize4D || src_dim > kShapeSize5D || src_dim != weight_dim) { in InitKernel()
3871 weight_dim = len(weight_shape)3872 if weight_dim != 1: