Home
last modified time | relevance | path

Searched refs:output_bytes_accessed (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc162 EXPECT_EQ(analysis.output_bytes_accessed(*root), sizeof(float) * 10 * 30); in TEST_F()
198 EXPECT_EQ(analysis.output_bytes_accessed(*root), sizeof(float) * 10 * 30); in TEST_F()
234 EXPECT_EQ(analysis.output_bytes_accessed(*root), sizeof(float) * 5 * 10 * 30); in TEST_F()
262 EXPECT_EQ(analysis.output_bytes_accessed(*root), in TEST_F()
284 EXPECT_EQ(analysis.output_bytes_accessed(*root), sizeof(float) * 10); in TEST_F()
318 EXPECT_EQ(analysis.output_bytes_accessed(*root), sizeof(float) * 8 * 18); in TEST_F()
400 EXPECT_EQ(analysis.output_bytes_accessed(*root), in TEST_F()
425 EXPECT_EQ(analysis.output_bytes_accessed(*root), sizeof(float) * 10); in TEST_F()
449 EXPECT_EQ(analysis.output_bytes_accessed(*root), sizeof(float) * 2 * 4); in TEST_F()
484 EXPECT_EQ(analysis.output_bytes_accessed(*root), sizeof(float) * 4); in TEST_F()
[all …]
Dhlo_cost_analysis.h164 int64 output_bytes_accessed(const HloInstruction& hlo,
Dhlo_cost_analysis.cc1090 int64 HloCostAnalysis::output_bytes_accessed(const HloInstruction& hlo, in output_bytes_accessed() function in xla::HloCostAnalysis
1130 bytes_written += output_bytes_accessed(hlo, indexed_shape.index); in GetBytesWritten()
Dmemory_space_assignment.cc281 float output_bytes_accessed = in GetInstructionElapsedDueToMemory() local
282 cost_analysis_.output_bytes_accessed(instruction); in GetInstructionElapsedDueToMemory()
284 output_bytes_accessed / alternate_mem_bandwidth_bytes_per_second_; in GetInstructionElapsedDueToMemory()
285 bytes_accessed -= output_bytes_accessed; in GetInstructionElapsedDueToMemory()