Searched refs:GetShapeSize (Results 1 – 2 of 2) sorted by relevance
52 float bytes_accessed = GetShapeSize(hlo->shape()); in Preprocess()54 bytes_accessed += GetShapeSize(operand->shape()); in Preprocess()125 int64 HloCostAnalysis::GetShapeSize(const Shape& shape) const { in GetShapeSize() function in xla::HloCostAnalysis192 current_properties_[kBytesAccessedKey] = GetShapeSize(slice->shape()) * 2; in HandleSlice()199 GetShapeSize(dynamic_slice->shape()) * 2; in HandleDynamicSlice()206 GetShapeSize(dynamic_update_slice->operand(1)->shape()) * 2; in HandleDynamicUpdateSlice()215 current_properties_[kBytesAccessedKey] = GetShapeSize(tuple->shape()); in HandleTuple()551 float bytes_accessed = GetShapeSize(hlo->operand(0)->shape()) / 2.0f; in HandleTriangularSolve()552 bytes_accessed += GetShapeSize(hlo->operand(1)->shape()); in HandleTriangularSolve()565 float bytes_accessed = GetShapeSize(hlo->operand(0)->shape()) / 2.0f; in HandleCholesky()[all …]
197 int64 GetShapeSize(const Shape& shape) const;