Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc160 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 0), sizeof(float) * 10 * 5); in TEST_F()
161 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 1), sizeof(float) * 5 * 30); in TEST_F()
194 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 0), in TEST_F()
196 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 1), in TEST_F()
230 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 0), in TEST_F()
232 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 1), in TEST_F()
260 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 0), sizeof(float) * 10 * 5); in TEST_F()
261 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 1), sizeof(float) * 5 * 30); in TEST_F()
283 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 0), sizeof(float) * 10); in TEST_F()
316 EXPECT_EQ(analysis.operand_bytes_accessed(*root, 0), sizeof(float) * 10 * 20); in TEST_F()
[all …]
Dhlo_cost_analysis.h162 int64 operand_bytes_accessed(const HloInstruction& hlo, int64 operand_num,
Dhlo_cost_analysis.cc1083 int64 HloCostAnalysis::operand_bytes_accessed(const HloInstruction& hlo, in operand_bytes_accessed() function in xla::HloCostAnalysis
1113 operand_bytes_accessed(hlo, operand_number, indexed_shape.index); in GetBytesRead()
Dmemory_space_assignment.cc268 float operand_bytes_accessed = cost_analysis_.operand_bytes_accessed( in GetInstructionElapsedDueToMemory() local
271 operand_bytes_accessed / alternate_mem_bandwidth_bytes_per_second_; in GetInstructionElapsedDueToMemory()
272 bytes_accessed -= operand_bytes_accessed; in GetInstructionElapsedDueToMemory()