Searched defs:shape_dims (Results 1 – 6 of 6) sorted by relevance
122 TfLiteIntArray* shape_dims = IntArrayFromInts(shape_dims_data); in TestReshape() local143 TfLiteIntArray* shape_dims = IntArrayFromInts(shape_dims_data); in TestReshapeQuantized() local165 const int shape_dims[] = {1, 2}; in TF_LITE_MICRO_TEST() local181 const int shape_dims[] = {1, 9}; in TF_LITE_MICRO_TEST() local197 const int shape_dims[] = {1, 4}; in TF_LITE_MICRO_TEST() local239 const int shape_dims[] = {1, 3}; in TF_LITE_MICRO_TEST() local271 const int shape_dims[] = {1, 3}; in TF_LITE_MICRO_TEST() local303 const int shape_dims[] = {0}; in TF_LITE_MICRO_TEST() local343 TfLiteIntArray* shape_dims = IntArrayFromInts(shape_dims_data); in TF_LITE_MICRO_TEST() local
55 size_t shape_dims = 1 + (data[data_ix++] % kMaxShapeDims); in FuzzImpl() local
191 llvm::SmallVector<int64_t, 4> shape_dims; in ConvertToMlirTensorType() local
115 auto shape_dims = AsInt64Slice(blocks_shape.dimensions()); in DiagonalBlocks() local
641 auto* shape_dims = shape->mutable_dims(); in ExtendShape() local652 std::vector<int>& shape_dims = *shape->mutable_dims(); in UnextendShape() local
179 const std::vector<int32> shape_dims = {static_cast<int32>(vals.size())}; in MakeConstNodeDef() local