Searched defs:dims_data (Results 1 – 5 of 5) sorted by relevance
56 RuntimeShape(int dimensions_count, const int32_t* dims_data) : size_(0) { in RuntimeShape()120 inline void ReplaceWith(int dimensions_count, const int32_t* dims_data) { in ReplaceWith()156 const int* dims_data = reinterpret_cast<const int*>(DimsData()); in FlatSize() local219 const int* dims_data = reinterpret_cast<const int*>(shape.DimsDataUpTo5D()); in Offset() local234 const int* dims_data = reinterpret_cast<const int*>(shape.DimsDataUpTo5D()); in Offset() local
41 const int32_t* dims_data = reinterpret_cast<const int32_t*>(dims->data); in GetTensorShape() local
505 const auto* dims_data = shape.DimsData(); in FlatSizeSkipDim() local
43 std::initializer_list<dims_type> dims_data, in FillOpModel()80 std::initializer_list<dims_type> dims_data, in QuantizedFillOpModel()
59 const int32_t* dims_data = input_shape.DimsData(); in PerChannelDequantize() local