Searched defs:output_permutation (Results 1 – 11 of 11) sorted by relevance
13 std::vector<int64_t> output_permutation(input_permutation.size(), -1); in InversePermutation() local
37 std::vector<int64_t> output_permutation(input_permutation.size(), -1); in InversePermutation() local
80 std::vector<int64_t> output_permutation(1 + // start index dimension. in ExpandInstruction() local
105 Tensor* output_permutation = nullptr; in Compute() local
245 std::vector<int> output_permutation(output_labels.size()); in Compute() local
561 std::vector<int> output_permutation(results_.size()); in BuildFunctionDef() local1202 std::vector<int>* output_permutation, NodeDef* node) { in Run()
117 std::vector<int>* output_permutation, in RewriteSubgraph()
799 NodeDef* call_def) { in TEST()844 NodeDef* call_def) { in TEST()
2206 std::vector<int>* output_permutation, NodeDef* node_def) { in operator ()()
80 std::vector<int>* output_permutation, in RewriteSubgraph()
354 SmallVector<int64_t, 4> output_permutation = { in matchAndRewrite() local