Searched defs:dims_data (Results 1 – 12 of 12) 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
43 int64_t dims_data[1] = {-1}; in TEST_F() local57 int64_t dims_data[2] = {-1, 0}; in TEST_F() local
48 int64_t dims_data[2] = {0, 1}; in TEST_F() 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()
148 std::vector<T> dims_data; in flatbufferDimsToVector() local227 std::vector<size_t> dims_data = flatbufferDimsToVector(tensor_value->dims()); in defineTensor() local1069 std::vector<size_t> dims_data = flatbufferDimsToVector(graph_node->perm()); in defineStaticTransposeNode() local1214 std::vector<size_t> dims_data = in defineStaticReshapeNode() local
48 std::vector<size_t> dims_data; in compileModel() local
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