Searched refs:ReLUAttributes (Results 1 – 11 of 11) 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()
38 const auto& attr = absl::any_cast<const ReLUAttributes&>(ctx.op_attr); in GenerateCode()
33 ReLUAttributes attr; in ReLUNoClipNoAlphaTest()62 ReLUAttributes attr; in ReLUClipTest()91 ReLUAttributes attr; in ReLUAlphaTest()120 ReLUAttributes attr; in ReLUAlphaClipTest()
28 const ReLUAttributes& attr);
24 const ReLUAttributes& attr) { in CreateReLU()
33 std::unique_ptr<GPUOperation> SelectReLU(const ReLUAttributes& attr,
54 std::unique_ptr<GPUOperation> SelectReLU(const ReLUAttributes& attr, in SelectReLU()
465 auto attr = absl::any_cast<ReLUAttributes>(node.operation.attributes); in GPUOperationFromNode()
357 struct ReLUAttributes { struct
415 ReLUAttributes attr; in MaybeFuseActivation()
1477 ReLUAttributes attr; in Parse()