Searched refs:UnaryOpsComposition (Results 1 – 3 of 3) sorted by relevance
29 class UnaryOpsComposition; // forward declare kernel57 friend class UnaryOpsComposition<T>;79 class UnaryOpsComposition : public OpKernel { class81 using Kernel = UnaryOpsComposition<T>;92 explicit UnaryOpsComposition(OpKernelConstruction* context) in UnaryOpsComposition() function in tensorflow::UnaryOpsComposition425 UnaryOpsComposition<T>);
2974 class UnaryOpsComposition : public ArithmeticOptimizerStage { class2976 explicit UnaryOpsComposition(const GraphOptimizerContext& ctx, in UnaryOpsComposition() function in tensorflow::grappler::__anon327bfa1e0111::UnaryOpsComposition3017 ~UnaryOpsComposition() override = default;3729 pipeline.AddStage<UnaryOpsComposition>(ctx, ctx_ext); in SimplifyArithmeticOps()
3764 TEST_F(ArithmeticOptimizerTest, UnaryOpsComposition) { in TEST_F() argument