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::UnaryOpsComposition424 UnaryOpsComposition<T>);
2863 class UnaryOpsComposition : public ArithmeticOptimizerStage { class2865 explicit UnaryOpsComposition(const GraphOptimizerContext& ctx, in UnaryOpsComposition() function in tensorflow::grappler::__anonbb66c6470111::UnaryOpsComposition2906 ~UnaryOpsComposition() override = default;3591 pipeline.AddStage<UnaryOpsComposition>(ctx, ctx_ext); in SimplifyArithmeticOps()
3602 TEST_F(ArithmeticOptimizerTest, UnaryOpsComposition) { in TEST_F() argument