Searched refs:PredictGatherOrSlice (Results 1 – 2 of 2) sorted by relevance
141 Costs PredictGatherOrSlice(const OpContext& op_context) const;
256 {kGather, wrap(&OpLevelCostEstimator::PredictGatherOrSlice)}, in OpLevelCostEstimator()257 {kGatherV2, wrap(&OpLevelCostEstimator::PredictGatherOrSlice)}, in OpLevelCostEstimator()258 {kSlice, wrap(&OpLevelCostEstimator::PredictGatherOrSlice)}, in OpLevelCostEstimator()1332 Costs OpLevelCostEstimator::PredictGatherOrSlice( in PredictGatherOrSlice() function in tensorflow::grappler::OpLevelCostEstimator