Searched refs:input_permutation (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | util.cc | 90 absl::Span<const int64> input_permutation) { in InversePermutation() argument
91 DCHECK(IsPermutation(input_permutation, input_permutation.size())); in InversePermutation()
92 std::vector<int64> output_permutation(input_permutation.size(), -1); in InversePermutation()
93 for (size_t i = 0; i < input_permutation.size(); ++i) { in InversePermutation()
94 output_permutation[input_permutation[i]] = i; in InversePermutation()
|
| D | util.h | 326 absl::Span<const int64> input_permutation);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 115 std::vector<int>* input_permutation, in RewriteSubgraph() argument
119 const int num_args = input_permutation->size(); in RewriteSubgraph()
168 (*input_permutation)[index] = i; in RewriteSubgraph()
|
| D | encapsulate_subgraphs_pass.h | 44 std::unique_ptr<Graph>* graph, std::vector<int>* input_permutation,
|
| D | extract_outside_compilation_pass.h | 56 std::vector<int>* input_permutation,
|
| D | encapsulate_subgraphs_pass.cc | 1087 std::vector<int> input_permutation(args_by_src_.size()); in BuildFunctionDef() local
1088 std::iota(input_permutation.begin(), input_permutation.end(), 0); in BuildFunctionDef()
1093 rewrite_subgraph_fn(arg_source_tensors, &graph_, &input_permutation, in BuildFunctionDef()
1099 if (input_permutation.size() != args_by_src_.size()) { in BuildFunctionDef()
1106 arg.second = input_permutation[arg.second]; in BuildFunctionDef()
1109 arg.second = input_permutation[arg.second]; in BuildFunctionDef()
2578 std::vector<int>* input_permutation, in Run()
2609 const int num_args = input_permutation->size(); in Run()
2640 (*input_permutation)[i] = const_pos; in Run()
2643 (*input_permutation)[i] = resource_pos; in Run()
[all …]
|
| D | encapsulate_subgraphs_pass_test.cc | 804 std::vector<int>* input_permutation, in TEST()
849 std::vector<int>* input_permutation, in TEST()
|
| D | extract_outside_compilation_pass.cc | 1483 std::unique_ptr<Graph>* graph, std::vector<int>* input_permutation, in operator ()() argument
|