- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for AnalyzeBlock (0.49 sec)
-
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc
auto type = mlir::cast<mlir::TensorType>(arg.getType()); if (type.hasRank()) { max_arg_size_ = std::max(max_arg_size_, GetRankedTensorSize(type)); } } } void CostAnalysis::AnalyzeBlock(mlir::Block* block) { for (auto& op : *block) { EvaluateCost(&op); } } void CostAnalysis::EvaluateCost(mlir::Operation* op) { if (auto cost_function =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0)