Searched defs:fused_op (Results 1 – 7 of 7) sorted by relevance
48 auto fused_op = rewriter.create<MapAndBatchDatasetOp>( in matchAndRewrite() local
202 Value fused_op = rewriter.create<FusedOpT>(fused_loc, result_type, in matchAndRewrite() local
290 std::unique_ptr<HloInstruction> fused_op = in FuseMatrixBiasAdd() local348 std::unique_ptr<HloInstruction> fused_op = in FuseVectorBiasAdd() local
101 def _VerifyValues(self, model_fn, use_low_precision, fused_op, epilog_ops): argument
220 Operation *fused_op = rewriter.create(*state); in matchAndRewrite() local300 Operation *fused_op = rewriter.create(*state); in matchAndRewrite() local
2220 NodeDef fused_op; in AddFusedContractionNode() local2269 NodeDef fused_op; in AddFusedContractionNode() local2688 NodeDef fused_op; in ReplaceMulMaximumWithLeakyRelu() local2741 NodeDef fused_op; in ReplaceSigmoidMulWithSwish() local2784 NodeDef fused_op; in AddFusedBatchNormExNode() local2853 NodeDef fused_op; in AddFusedBatchNormGradExNode() local3115 NodeDef fused_op; in AddTensorToHashBucketNode() local
2098 for (auto& fused_op : fused_op_contexts) { in PredictFusedOp() local