| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_profile_printer_data.proto | 32 float flop_count = 4;
|
| human_readable_profile_builder.h | 47 tensorflow::StringPiece category, int64 cycles, int64 flop_count,
52 flop_count, transcendental_count, bytes_accessed, optimal_seconds});
64 int64 flop_count; member in struct:xla::HumanReadableProfileBuilder::OpInfo
|
| hlo_cost_analysis_test.cc | 160 EXPECT_EQ(analysis.flop_count(), 2 * 10 * 30 * 5);
181 EXPECT_EQ(analysis.flop_count(), 10);
206 EXPECT_EQ(analysis.flop_count(), 8 * 18 * 2 * 3 * 3);
228 EXPECT_EQ(analysis.flop_count(), 10 * 20 - 10);
245 EXPECT_EQ(analysis.flop_count(), 2 * 4 * (4 * 5 - 1));
266 EXPECT_EQ(analysis.flop_count(), 2 * 4 * (4 * 5 - 1 + 1));
276 EXPECT_EQ(analysis.flop_count(), 0);
299 EXPECT_EQ(analysis.flop_count(), 2 * 1000 + 200 + 3 * 200);
330 EXPECT_EQ(conv_analysis.flop_count(), matmul_analysis.flop_count());
[all...] |
| hlo_profile_printer.cc | 53 instruction_info.flop_count(),
|
| human_readable_profile_builder.cc | 66 op.flop_count <= 0
68 : HumanReadableNumFlops(op.flop_count, nsecs).c_str(),
82 total_flops += op.flop_count;
|
| hlo_execution_profile.cc | 94 instruction_info->set_flop_count(cost_analysis.flop_count(*hlo));
|
| hlo_cost_analysis.h | 122 float flop_count() const;
129 int64 flop_count(const HloInstruction& hlo) const;
|
| hlo_cost_analysis.cc | 545 float HloCostAnalysis::flop_count() const { function in class:xla::HloCostAnalysis
561 int64 HloCostAnalysis::flop_count(const HloInstruction& hlo) const { function in class:xla::HloCostAnalysis
|
| service.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| parallel_task_assignment.cc | 68 cost_analysis_->flop_count(*instruction) /
88 1 * cost_analysis_->flop_count(*instruction) +
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| fusion_merger.cc | 113 return bytes > 0.0 ? analysis.flop_count() / bytes : analysis.flop_count();
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| client.cc | 385 computation_stats.flop_count() + computation_stats.transcendental_count();
391 "[Execution Statistics] flop count: ", computation_stats.flop_count(),
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| xla_data.proto | 208 double flop_count = 1;
|
