Searched defs:MklSoftmaxPrimitive (Results 1 – 1 of 1) sorted by relevance
52 class MklSoftmaxPrimitive : public MklPrimitive { class54 explicit MklSoftmaxPrimitive(const MklSoftmaxParams& fwdParams) in MklSoftmaxPrimitive() function in tensorflow::MklSoftmaxPrimitive