Searched refs:PredictSoftmax (Results 1 – 2 of 2) sorted by relevance
181 Status PredictSoftmax(const OpContext& op_context,
523 wrap(&OpLevelCostEstimator::PredictSoftmax)); in OpLevelCostEstimator()2484 Status OpLevelCostEstimator::PredictSoftmax(const OpContext& op_context, in PredictSoftmax() function in tensorflow::grappler::OpLevelCostEstimator