Home
last modified time | relevance | path

Searched defs:input_permutation (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dpermutation_util.cc35 absl::Span<const int64_t> input_permutation) { in InversePermutation()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc116 std::vector<int>* input_permutation, in RewriteSubgraph()
Dencapsulate_subgraphs_pass.cc559 std::vector<int> input_permutation(args_by_src_.size()); in BuildFunctionDef() local
1202 std::vector<int>* output_permutation, NodeDef* node) { in Run()
Dencapsulate_subgraphs_pass_test.cc799 NodeDef* call_def) { in TEST()
844 NodeDef* call_def) { in TEST()
Dextract_outside_compilation_pass.cc2205 std::unique_ptr<Graph>* graph, std::vector<int>* input_permutation, in operator ()()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc79 std::vector<int>* input_permutation, in RewriteSubgraph()