Searched defs:dims_data (Results 1 – 8 of 8) 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
521 const auto* dims_data = shape.DimsData(); in FlatSizeSkipDim() local
42 const int32_t* dims_data = shape.data(); in QuantizeInt8PerChannel() local79 const int32_t* dims_data = shape.data(); in GetInt8QuantizationScalePerChannel() 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
66 const int32_t* dims_data = input_shape.DimsData(); in PerChannelQuantize() local
46 const int* dims_data = in BroadcastBinaryFunction4DSlow() local