Searched defs:dims_data (Results 1 – 5 of 5) sorted by relevance
41 const int32_t* dims_data = reinterpret_cast<const int32_t*>(dims->data); in GetTensorShape() local
175 RuntimeShape(int dimensions_count, const int32_t* dims_data) : size_(0) { in RuntimeShape()255 inline void ReplaceWith(int dimensions_count, const int32_t* dims_data) { in ReplaceWith()291 const int* dims_data = reinterpret_cast<const int*>(DimsData()); in FlatSize() local405 const int* dims_data = reinterpret_cast<const int*>(shape.DimsDataUpTo5D()); in Offset() local416 const int* dims_data = reinterpret_cast<const int*>(shape.DimsDataUpTo5D()); in Offset() local676 const auto* dims_data = shape.DimsData(); in FlatSizeSkipDim() local
36 const int32_t* dims_data = reinterpret_cast<const int32_t*>(dims->data); in GetTensorShape() local
43 std::initializer_list<dims_type> dims_data, in FillOpModel()81 std::initializer_list<dims_type> dims_data, in QuantizedFillOpModel()
59 const int32_t* dims_data = input_shape.DimsData(); in PerChannelDequantize() local