MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / get_int8_chwn4_tensorcore_args

Function get_int8_chwn4_tensorcore_args

dnn/test/common/conv_bias.cpp:752–758  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

750}
751
752std::vector<TestArg> get_int8_chwn4_tensorcore_args(size_t kernel_size) {
753 auto args = get_int8_nchw4_tensorcore_args(kernel_size);
754 for (auto& arg : args) {
755 convert_arg_from_nchw4_to_chwn4(arg);
756 }
757 return args;
758}
759
760void check_conv_bias(
761 DType src_dtype, DType filter_dtype, DType bias_dtype, DType dst_dtype,

Callers 1

TEST_FFunction · 0.85

Tested by

no test coverage detected