Searched defs:TransposeContext (Results 1 – 3 of 3) sorted by relevance
49 struct TransposeContext { struct58 static Status InitializeTransposeContext(const GrapplerItem& item, in InitializeTransposeContext() argument69 FrameView frames;70 GraphDef graph;74 int num_nodes;75 absl::flat_hash_set<string> nodes_to_preserve;76 std::unique_ptr<GraphProperties> graph_properties;77 std::unique_ptr<utils::MutableGraphView> graph_view;79 string target_device;80 string src_format;[all …]
69 Status TransposeNode(TransposeContext*, utils::MutableNodeView*) override { in TransposeNode() argument
37 struct TransposeContext { struct38 TransposeContext(TfLiteContext* context, TfLiteNode* node) { in TransposeContext() argument43 const TfLiteTensor* input;44 const TfLiteTensor* perm;45 TfLiteTensor* output;