Searched refs:no_gemm (Results 1 – 2 of 2) sorted by relevance
41 bool no_gemm; member
1047 Conv2DProcessor(const OptimizeContext& opt_cxt, bool no_gemm) in Conv2DProcessor() argument1048 : NodeProcessor(opt_cxt), no_gemm_(no_gemm) {} in Conv2DProcessor()1116 Conv2DBackpropFilterProcessor(const OptimizeContext& opt_cxt, bool no_gemm) in Conv2DBackpropFilterProcessor() argument1117 : Conv2DProcessor(opt_cxt, no_gemm) {} in Conv2DBackpropFilterProcessor()1137 Conv2DBackpropInputProcessor(const OptimizeContext& opt_cxt, bool no_gemm) in Conv2DBackpropInputProcessor() argument1138 : Conv2DProcessor(opt_cxt, no_gemm) {} in Conv2DBackpropInputProcessor()2018 node_processor.reset(new Conv2DProcessor(opt_cxt, config_.no_gemm)); in Expand()2021 new Conv2DBackpropFilterProcessor(opt_cxt, config_.no_gemm)); in Expand()2024 new Conv2DBackpropInputProcessor(opt_cxt, config_.no_gemm)); in Expand()2222 config.no_gemm = true; in Optimize()