Searched defs:softmax_size (Results 1 – 2 of 2) sorted by relevance
129 int softmax_size() const override { return (SafeGetNumLayers() > 0) ? 1 : 0; } in softmax_size() function
825 const std::string softmax_size = src_op.outputs[0] + "/softmax_insert_size"; in ConvertSoftmaxOperator() local866 const std::string softmax_size = in ConvertLogSoftmaxOperator() local