Searched defs:fused_op (Results 1 – 5 of 5) sorted by relevance
44 auto fused_op = rewriter.create<MapAndBatchDatasetOp>( in matchAndRewrite() local
62 } else if (FusedOp fused_op = operand.getDefiningOp<FusedOp>()) { in matchAndRewrite() local
175 Value fused_op = rewriter.create<FusedOpT>(fused_loc, result_type, in matchAndRewrite() local
1072 NodeDef fused_op; in IsCpuCompatibleDataType() local1120 NodeDef fused_op; in IsCpuCompatibleDataType() local1395 NodeDef fused_op; in IsCpuCompatibleDataType() local
2064 for (auto& fused_op : fused_op_contexts) { in PredictFusedOp() local