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
|