Searched refs:ReLUAttributes (Results 1 – 12 of 12) sorted by relevance
47 ReLUAttributes attr; in TEST_F()60 ReLUAttributes attr; in TEST_F()73 ReLUAttributes attr; in TEST_F()86 ReLUAttributes attr; in TEST_F()
41 const auto& attr = std::any_cast<const ReLUAttributes&>(ctx.op_attr); in GenerateCode()
34 ReLUAttributes attr; in ReLUNoClipNoAlphaTest()63 ReLUAttributes attr; in ReLUClipTest()92 ReLUAttributes attr; in ReLUAlphaTest()121 ReLUAttributes attr; in ReLUAlphaClipTest()
27 ElementwiseDescriptor CreateReLU(const ReLUAttributes& attr,31 const ReLUAttributes& attr);
26 ElementwiseDescriptor CreateReLU(const ReLUAttributes& attr, in CreateReLU()55 const ReLUAttributes& attr) { in CreateReLU()
70 void AddReluNode(const ReLUAttributes& attr);356 ReLUAttributes* attr = in AddNode()357 absl::any_cast<ReLUAttributes>(&node->operation.attributes); in AddNode()490 void ThinPointwiseFuser::AddReluNode(const ReLUAttributes& attr) { in AddReluNode()
33 std::unique_ptr<GPUOperation> SelectReLU(const ReLUAttributes& attr,
62 std::unique_ptr<GPUOperation> SelectReLU(const ReLUAttributes& attr, in SelectReLU()
631 auto attr = absl::any_cast<ReLUAttributes>(node.operation.attributes); in GPUOperationFromNodePart0()
398 struct ReLUAttributes { struct
399 ReLUAttributes attr; in MaybeFuseActivation()
407 ReLUAttributes relu_attr; in Parse()1836 ReLUAttributes attr; in Parse()