Searched refs:input_permutation (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | permutation_util.cc | 35 absl::Span<const int64> input_permutation) { in InversePermutation() argument
36 DCHECK(IsPermutation(input_permutation)); in InversePermutation()
37 std::vector<int64> output_permutation(input_permutation.size(), -1); in InversePermutation()
38 for (size_t i = 0; i < input_permutation.size(); ++i) { in InversePermutation()
39 output_permutation.at(input_permutation.at(i)) = i; in InversePermutation()
|
| D | permutation_util.h | 74 absl::Span<const int64> input_permutation);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 583 std::vector<int> input_permutation(args_by_src_.size()); in BuildFunctionDef() local
584 std::iota(input_permutation.begin(), input_permutation.end(), 0); in BuildFunctionDef()
589 rewrite_subgraph_fn(arg_source_tensors, &graph_, &input_permutation, in BuildFunctionDef()
595 if (input_permutation.size() != args_by_src_.size()) { in BuildFunctionDef()
602 arg.second = input_permutation[arg.second]; in BuildFunctionDef()
605 arg.second = input_permutation[arg.second]; in BuildFunctionDef()
1215 std::vector<int>* input_permutation, in Run()
1249 const int num_args = input_permutation->size(); in Run()
1280 (*input_permutation)[i] = const_pos; in Run()
1283 (*input_permutation)[i] = resource_pos; in Run()
[all …]
|
| D | encapsulate_subgraphs_pass.h | 53 std::unique_ptr<Graph>* graph, std::vector<int>* input_permutation,
|
| D | extract_outside_compilation_pass.h | 57 std::vector<int>* input_permutation,
|
| D | encapsulate_xla_computations_pass.cc | 111 std::vector<int>* input_permutation, in RewriteSubgraph() argument
115 const int num_args = input_permutation->size(); in RewriteSubgraph()
164 (*input_permutation)[index] = i; in RewriteSubgraph()
|
| D | encapsulate_subgraphs_pass_test.cc | 797 std::vector<int>* input_permutation, in TEST()
842 std::vector<int>* input_permutation, in TEST()
|
| D | extract_outside_compilation_pass.cc | 2204 std::unique_ptr<Graph>* graph, std::vector<int>* input_permutation, in operator ()() argument
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 78 std::vector<int>* input_permutation, in RewriteSubgraph() argument
127 const int num_args = input_permutation->size(); in RewriteSubgraph()
226 (*input_permutation)[index] = i; in RewriteSubgraph()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOps.td | 94 Optionally, can take `input_permutation` and `output_permutation` attributes
|