1 /* Copyright 2017 The TensorFlow Authors. All Rights Reserved. 2 3 Licensed under the Apache License, Version 2.0 (the "License"); 4 you may not use this file except in compliance with the License. 5 You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9 Unless required by applicable law or agreed to in writing, software 10 distributed under the License is distributed on an "AS IS" BASIS, 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 See the License for the specific language governing permissions and 13 limitations under the License. 14 ==============================================================================*/ 15 #ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTERUTILS_H 16 #define TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTERUTILS_H 17 18 #include <string> 19 20 namespace toco { 21 22 // Check if string x includes string search_pattern. 23 bool StrContains(const string& x, const string& search_pattern); 24 25 // Transpose a 2D tensor of size row * col pointed by "tensor" and return the 26 // results in "transposed_tensor". "transposed_tensor" must be pre-allocated 27 // by the same size as "tensor". 28 void Transpose2DTensor(const float* tensor, int row, int col, 29 float* transposed_tensor); 30 31 } // end namespace toco 32 33 #endif // TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTERUTILS_H 34
