HomeSort by relevance Sort by last modified time
    Searched refs:GetShapeSize (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_cost_analysis.cc 52 float bytes_accessed = GetShapeSize(hlo->shape());
54 bytes_accessed += GetShapeSize(operand->shape());
125 int64 HloCostAnalysis::GetShapeSize(const Shape& shape) const {
192 current_properties_[kBytesAccessedKey] = GetShapeSize(slice->shape()) * 2;
199 GetShapeSize(dynamic_slice->shape()) * 2;
206 GetShapeSize(dynamic_update_slice->operand(1)->shape()) * 2;
215 current_properties_[kBytesAccessedKey] = GetShapeSize(tuple->shape());
551 float bytes_accessed = GetShapeSize(hlo->operand(0)->shape()) / 2.0f;
552 bytes_accessed += GetShapeSize(hlo->operand(1)->shape());
565 float bytes_accessed = GetShapeSize(hlo->operand(0)->shape()) / 2.0f
    [all...]
hlo_cost_analysis.h 197 int64 GetShapeSize(const Shape& shape) const;

Completed in 89 milliseconds