- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for op_costs (0.1 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/runtime_metadata.fbs
// ["CPU", "GPU"] // } // // subgraph_metadata: { // [ // op1: // hardware: 0 // op_costs: [10, -1], // op2: // hardware: 1 // op_costs: [20, 5], // op3: // hardware: 1 // op_costs: [30, 10], // op4: // hardware: 0 // op_costs: [40, -1], // ] // } // } table HardwareMetadata { // List of different hardwares this model can use.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 21 01:22:53 UTC 2021 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter_test.cc
expected_op_metadata->hardware()); EXPECT_EQ(result_op_metadata->op_costs()->size(), expected_op_metadata->op_costs()->size()); for (int i = 0; i < result_op_metadata->op_costs()->size(); ++i) { EXPECT_FLOAT_EQ(result_op_metadata->op_costs()->Get(i), expected_op_metadata->op_costs()->Get(i)); } } } } TEST(ExporterTest, Valid) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 06:11:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc
void PrintOpStatsPass::CountOp( DenseMap<StringRef, llvm::StringMap<int64_t>> &op_count_map, StringRef op_name, StringRef dtype) { auto &op_counts = op_count_map[op_name]; if (auto it = op_counts.find(dtype); it != op_counts.end()) { it->second++; } else { op_counts[dtype] = 1; } } void PrintOpStatsPass::PrintSummary() { *os_ << "Summary on the non-converted ops:\n";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0)