Searched refs:dimension_ (Results 1 – 7 of 7) sorted by relevance
170 void SetDimensions(const size_t dimension) { dimension_ = dimension; } in SetDimensions()183 dimension_ = dimension; in SetTfLayout()205 CHECK(dimension == dimension_); in SetTfDimOrder()215 CHECK(dimension == dimension_); in SetTfDimOrder()226 CHECK(dimension == dimension_); in SetTfDimOrder()241 size_t GetDimension() const { return dimension_; } in GetDimension()352 dimension_ = *(reinterpret_cast<const size_t*>(buf + DIMS_OFFSET)); in DeSerializeMklShape()353 CHECK(buf_size >= SIZE_OF_MKL_SERIAL_DATA(dimension_)) in DeSerializeMklShape()355 sizes_ = new size_t[dimension_]; in DeSerializeMklShape()356 strides_ = new size_t[dimension_]; in DeSerializeMklShape()[all …]
325 TestParameters (Type wsiType_, TestDimension dimension_) in TestParameters()327 , dimension (dimension_) in TestParameters()
340 TestParameters (Type wsiType_, TestDimension dimension_) in TestParameters()342 , dimension (dimension_) in TestParameters()
208 TestParameters (vk::wsi::Type wsiType_, TestDimension dimension_) in TestParameters()210 , dimension (dimension_) in TestParameters()
1551 int64 dimension() const { return dimension_; } in dimension()1567 int64 dimension_; variable
2641 dimension_(dimension) { in HloGetDimensionSizeInstruction()
458 Expression dimension_ = cloneNodes(_n.getDimension(), _arg); in visit() local460 ArrayCreationLevel r = new ArrayCreationLevel(_n.getRange(), dimension_, ann); in visit()