Searched refs:output_permutation (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | permutation_util.cc | 37 std::vector<int64> output_permutation(input_permutation.size(), -1); in InversePermutation() local
39 output_permutation.at(input_permutation.at(i)) = i; in InversePermutation()
41 return output_permutation; in InversePermutation()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| 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()
|
| D | einsum_op_impl.h | 698 std::vector<int> output_permutation(output_labels.size()); in Compute()
707 output_permutation[i] = label_to_position[output_labels[i]]; in Compute()
713 ctx, output_inflated, output_permutation, &output)); in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.h | 54 std::vector<int>* output_permutation, NodeDef* node_def)>
|
| D | extract_outside_compilation_pass.h | 58 std::vector<int>* output_permutation, NodeDef* node_def);
|
| D | encapsulate_xla_computations_pass.cc | 112 std::vector<int>* output_permutation, in RewriteSubgraph() argument
116 const int num_retvals = output_permutation->size(); in RewriteSubgraph()
174 (*output_permutation)[index] = i; in RewriteSubgraph()
|
| D | encapsulate_subgraphs_pass.cc | 585 std::vector<int> output_permutation(results_.size()); in BuildFunctionDef() local
586 std::iota(output_permutation.begin(), output_permutation.end(), 0); in BuildFunctionDef()
590 &output_permutation, &call_node_def_)); in BuildFunctionDef()
598 if (output_permutation.size() != results_.size()) { in BuildFunctionDef()
608 result.second = output_permutation[result.second]; in BuildFunctionDef()
1216 std::vector<int>* output_permutation, NodeDef* node) { in Run() argument
|
| D | encapsulate_subgraphs_pass_test.cc | 798 std::vector<int>* output_permutation, in TEST()
843 std::vector<int>* output_permutation, in TEST()
|
| D | extract_outside_compilation_pass.cc | 2205 std::vector<int>* output_permutation, NodeDef* node_def) { in operator ()() argument
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 79 std::vector<int>* output_permutation, in RewriteSubgraph() argument
128 const int num_retvals = output_permutation->size(); in RewriteSubgraph()
237 (*output_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
|