Searched refs:flop_count (Results 1 – 13 of 13) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_cost_analysis_test.cc | 151 EXPECT_EQ(analysis.flop_count(), 2 * 10 * 30 * 5); in TEST_F() 185 EXPECT_EQ(analysis.flop_count(), 2 * 10 * 30 * 5 * 5); in TEST_F() 221 EXPECT_EQ(analysis.flop_count(), 2 * 10 * 30 * 5 * 5); in TEST_F() 251 EXPECT_EQ(analysis.flop_count(), 2 * 10 * 30 * 5 * 5); in TEST_F() 278 EXPECT_EQ(analysis.flop_count(), 10); in TEST_F() 309 EXPECT_EQ(analysis.flop_count(), 8 * 18 * 2 * 3 * 3); in TEST_F() 341 EXPECT_EQ(analysis.flop_count(), 2 * kLarge); in TEST_F() 364 EXPECT_EQ(analysis.flop_count(), 2 * kLarge); in TEST_F() 389 EXPECT_EQ(analysis.flop_count(), 120 * 8 * 18 * 2 * 3 * 3); in TEST_F() 418 EXPECT_EQ(analysis.flop_count(), 10 * 20 - 10); in TEST_F() [all …]
|
D | human_readable_profile_builder.h | 49 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
|
D | human_readable_profile_builder.cc | 93 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()
|
D | hlo_profile_printer.cc | 60 instruction_info.flop_count(), in PrintHloProfile()
|
D | hlo_cost_analysis.h | 148 float flop_count() const; 159 int64 flop_count(const HloInstruction& hlo) const;
|
D | hlo_profile_printer_data.proto | 32 float flop_count = 4; field
|
D | hlo_execution_profile.cc | 103 instruction_info->set_flop_count(cost_analysis.flop_count(*hlo)); in CreateHloProfilePrinterData()
|
D | hlo_cost_analysis.cc | 1055 float HloCostAnalysis::flop_count() const { in flop_count() function in xla::HloCostAnalysis 1071 int64 HloCostAnalysis::flop_count(const HloInstruction& hlo) const { in flop_count() function in xla::HloCostAnalysis
|
D | service.cc | 1125 stats.set_flop_count(analysis.flop_count()); in GetComputationGraphStats()
|
D | memory_space_assignment.cc | 251 cost_analysis_.flop_count(instruction) / in GetInstructionElapsedDueToCompute()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | parallel_task_assignment.cc | 71 cost_analysis_->flop_count(*instruction) / in GetParallelTaskCount() 92 1 * cost_analysis_->flop_count(*instruction) + in GetParallelTaskCount()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | client.cc | 472 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/ |
D | xla_data.proto | 243 double flop_count = 1; field
|