Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc1022 int64 update_size = GetShapeSize(scatter->operand(2)->shape()); in HandleScatter() local
1024 update_size * 3 + GetShapeSize(scatter->operand(1)->shape()); in HandleScatter()
1025 SetOperandBytesAccessed(0, update_size); in HandleScatter()
1027 SetOperandBytesAccessed(2, update_size); in HandleScatter()
1028 SetOutputBytesAccessed(update_size); in HandleScatter()