Searched refs:GetShapeSize (Results 1 – 3 of 3) sorted by relevance
57 float bytes_accessed = GetShapeSize(hlo->shape()); in Preprocess()58 SetOutputBytesAccessed(GetShapeSize(hlo->shape())); in Preprocess()61 bytes_accessed += GetShapeSize(operand->shape()); in Preprocess()62 SetOperandBytesAccessed(i, GetShapeSize(operand->shape())); in Preprocess()138 int64 HloCostAnalysis::GetShapeSize(const Shape& shape) const { in GetShapeSize() function in xla::HloCostAnalysis160 size += GetShapeSize(user->shape()); in FusionParameterReadBytes()163 size += hlo == user->operand(0) ? GetShapeSize(user->shape()) in FusionParameterReadBytes()164 : GetShapeSize(hlo->shape()); in FusionParameterReadBytes()169 ? GetShapeSize(user->operand(1)->shape()) in FusionParameterReadBytes()170 : GetShapeSize(hlo->shape()); in FusionParameterReadBytes()[all …]
133 int64 GetShapeSize(const Shape& shape) const;
312 int64 size_in_bytes = cost_analysis_.GetShapeSize(shape); in GetAsyncCopyElapsed()