Searched refs:weight_format (Results 1 – 2 of 2) sorted by relevance
113 memory::format_tag weight_format = in Compute() local121 (this->is_weight_const_) ? memory::format_tag::any : weight_format, in Compute()225 memory::desc(weight_dims, MklDnnType<T>(), weight_format); in Compute()
52 MEMORY_FORMAT weight_format; member64 MEMORY_FORMAT weight_format = MEMORY_FORMAT::any,71 weight_format(weight_format), in src_dims()185 matmul_fwd_params.weight_format)); in Setup()341 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.weight_format); in CreateKey()