Searched refs:output_permutation (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lu_op.cc | 105 Tensor* output_permutation = nullptr; in Compute() local
107 &output_permutation)); in Compute()
108 outputs.emplace_back(output_permutation); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | util.cc | 92 std::vector<int64> output_permutation(input_permutation.size(), -1); in InversePermutation() local
94 output_permutation[input_permutation[i]] = i; in InversePermutation()
96 return output_permutation; in InversePermutation()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 116 std::vector<int>* output_permutation, in RewriteSubgraph() argument
120 const int num_retvals = output_permutation->size(); in RewriteSubgraph()
178 (*output_permutation)[index] = i; in RewriteSubgraph()
|
| D | encapsulate_subgraphs_pass.h | 45 std::vector<int>* output_permutation, NodeDef* node_def)>
|
| D | extract_outside_compilation_pass.h | 57 std::vector<int>* output_permutation, NodeDef* node_def);
|
| D | encapsulate_subgraphs_pass.cc | 1089 std::vector<int> output_permutation(results_.size()); in BuildFunctionDef() local
1090 std::iota(output_permutation.begin(), output_permutation.end(), 0); in BuildFunctionDef()
1094 &output_permutation, &call_node_def_)); in BuildFunctionDef()
1102 if (output_permutation.size() != results_.size()) { in BuildFunctionDef()
1112 result.second = output_permutation[result.second]; in BuildFunctionDef()
2579 std::vector<int>* output_permutation, NodeDef* node) { in Run() argument
|
| D | encapsulate_subgraphs_pass_test.cc | 805 std::vector<int>* output_permutation, in TEST()
850 std::vector<int>* output_permutation, in TEST()
|
| D | extract_outside_compilation_pass.cc | 1484 std::vector<int>* output_permutation, NodeDef* node_def) { in operator ()() argument
|