Searched defs:output_permutation (Results 1 – 7 of 7) sorted by relevance
37 std::vector<int64> output_permutation(input_permutation.size(), -1); in InversePermutation() local
105 Tensor* output_permutation = nullptr; in Compute() local
112 std::vector<int>* output_permutation, in RewriteSubgraph()
585 std::vector<int> output_permutation(results_.size()); in BuildFunctionDef() local1216 std::vector<int>* output_permutation, NodeDef* node) { in Run()
799 NodeDef* call_def) { in TEST()844 NodeDef* call_def) { in TEST()
2205 std::vector<int>* output_permutation, NodeDef* node_def) { in operator ()()
79 std::vector<int>* output_permutation, in RewriteSubgraph()