Home
last modified time | relevance | path

Searched refs:flop_count (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc152 EXPECT_EQ(analysis.flop_count(), 2 * 10 * 30 * 5); in TEST_F()
181 EXPECT_EQ(analysis.flop_count(), 2 * 10 * 30 * 5 * 5); in TEST_F()
210 EXPECT_EQ(analysis.flop_count(), 2 * 10 * 30 * 5 * 5); in TEST_F()
233 EXPECT_EQ(analysis.flop_count(), 2 * 10 * 30 * 5 * 5); in TEST_F()
254 EXPECT_EQ(analysis.flop_count(), 10); in TEST_F()
281 EXPECT_EQ(analysis.flop_count(), 8 * 18 * 2 * 3 * 3); in TEST_F()
310 EXPECT_EQ(analysis.flop_count(), 120 * 8 * 18 * 2 * 3 * 3); in TEST_F()
331 EXPECT_EQ(analysis.flop_count(), 10 * 20 - 10); in TEST_F()
348 EXPECT_EQ(analysis.flop_count(), 2 * 4 * (4 * 5 - 1)); in TEST_F()
368 EXPECT_EQ(analysis.flop_count(), 2 * 4 * (4 * 5 - 1 + 1)); in TEST_F()
[all …]
Dhuman_readable_profile_builder.h49 absl::string_view category, int64 cycles, int64 flop_count, in AddOp() argument
53 cycles, flop_count, transcendental_count, in AddOp()
66 int64 flop_count; // -1 if unknown member
Dhuman_readable_profile_builder.cc93 op.flop_count > 0 && nsecs > 0 in ToString()
94 ? HumanReadableNumFlops(op.flop_count, nsecs) in ToString()
117 total_flops += std::max(op.flop_count, int64{0}); in ToString()
Dhlo_cost_analysis.h133 float flop_count() const;
140 int64 flop_count(const HloInstruction& hlo) const;
Dhlo_profile_printer.cc60 instruction_info.flop_count(), in PrintHloProfile()
Dhlo_execution_profile.cc102 instruction_info->set_flop_count(cost_analysis.flop_count(*hlo)); in CreateHloProfilePrinterData()
Dhlo_profile_printer_data.proto32 float flop_count = 4; field
Dhlo_cost_analysis.cc745 float HloCostAnalysis::flop_count() const { in flop_count() function in xla::HloCostAnalysis
761 int64 HloCostAnalysis::flop_count(const HloInstruction& hlo) const { in flop_count() function in xla::HloCostAnalysis
Dservice.cc1148 stats.set_flop_count(analysis.flop_count()); in GetComputationGraphStats()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc70 cost_analysis_->flop_count(*instruction) / in GetParallelTaskCount()
90 1 * cost_analysis_->flop_count(*instruction) + in GetParallelTaskCount()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc472 computation_stats.flop_count() + computation_stats.transcendental_count(); in ExecutionStatsAsString()
478 "[Execution Statistics] flop count: ", computation_stats.flop_count(), in ExecutionStatsAsString()
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto233 double flop_count = 1; field