- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for RecordCost (0.08 sec)
-
tensorflow/compiler/mlir/tfrt/tests/analysis/update_op_cost_in_tfrt_mlir_test.cc
for (auto& [op_key, cost] : expected_op_cost_map) { cost = rand_r(&seed) % 1000; } tensorflow::tfrt_stub::CostRecorder cost_recorder; for (const auto& [op_key, cost] : expected_op_cost_map) { cost_recorder.RecordCost(op_key, cost); } // Update the TFRT MLIR with the cost recorder. tfrt_compiler::UpdateOpCostInTfrtMlir(module.get(), cost_recorder); // Check the updated costs.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 22:52:12 UTC 2023 - 3.1K bytes - Viewed (0)