Searched defs:input_b (Results 1 – 3 of 3) sorted by relevance
239 void MatMulCPUKernel::LaunchARM(const float *input_a, const float *input_b, float *output) { in LaunchARM()246 void MatMulCPUKernel::LaunchX64(const float *input_a, const float *input_b, float *output) const { in LaunchX64() argument263 const auto input_b = reinterpret_cast<float *>(inputs[1]->addr); in Launch() local
316 auto* input_b = Structure( in TEST_F() local786 auto* input_b = Structure( in TEST_F() local
1704 auto input_b = node->input(kTwoNum); // conv bias in ExportMergeConv() local1716 auto input_b = node->input(kTwoNum); // matmul bias in ExportMergeGemm() local