Searched refs:Conv2DProcessor (Results 1 – 1 of 1) sorted by relevance
1045 class Conv2DProcessor : public NodeProcessor { class1047 Conv2DProcessor(const OptimizeContext& opt_cxt, bool no_gemm) in Conv2DProcessor() function in tensorflow::grappler::__anonbd4a17fb0111::Conv2DProcessor1114 class Conv2DBackpropFilterProcessor : public Conv2DProcessor {1117 : Conv2DProcessor(opt_cxt, no_gemm) {} in Conv2DBackpropFilterProcessor()1123 return Conv2DProcessor::IsGemmUsed(filter_shape, input_shape); in IsGemmUsed()1135 class Conv2DBackpropInputProcessor : public Conv2DProcessor {1138 : Conv2DProcessor(opt_cxt, no_gemm) {} in Conv2DBackpropInputProcessor()1144 return Conv2DProcessor::IsGemmUsed(filter_shape, input_shape); in IsGemmUsed()2018 node_processor.reset(new Conv2DProcessor(opt_cxt, config_.no_gemm)); in Expand()2026 node_processor.reset(new Conv2DProcessor(opt_cxt, true)); in Expand()