Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for op_costs (0.1 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top