Searched defs:TransposeContext (Results 1 – 4 of 4) sorted by relevance
50 struct TransposeContext { struct63 FrameView frames;64 GraphDef graph;68 int num_nodes;69 absl::flat_hash_set<string> nodes_to_preserve;70 std::unique_ptr<GraphProperties> graph_properties;71 std::unique_ptr<utils::MutableGraphView> graph_view;72 std::unique_ptr<const VirtualPlacer> virtual_placer;74 string target_device;75 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;