Searched refs:PredictEinsum (Results 1 – 2 of 2) sorted by relevance
175 Status PredictEinsum(const OpContext& op_context,
407 wrap(&OpLevelCostEstimator::PredictEinsum)); in OpLevelCostEstimator()409 wrap(&OpLevelCostEstimator::PredictEinsum)); in OpLevelCostEstimator()1801 Status OpLevelCostEstimator::PredictEinsum(const OpContext& op_context, in PredictEinsum() function in tensorflow::grappler::OpLevelCostEstimator