Searched refs:SoftmaxAttributes (Results 1 – 4 of 4) sorted by relevance
46 SoftmaxAttributes attr; in TEST()68 SoftmaxAttributes attr; in TEST()88 SoftmaxAttributes attr; in TEST()108 SoftmaxAttributes attr; in TEST()135 SoftmaxAttributes attr; in TEST()170 SoftmaxAttributes attr; in TEST()
46 const auto& attr = absl::any_cast<const SoftmaxAttributes&>(ctx.op_attr); in GenerateCode()
379 struct SoftmaxAttributes { struct
1777 SoftmaxAttributes attr; in Parse()