Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc115 std::vector<int>* input_permutation, in RewriteSubgraph()
Dencapsulate_subgraphs_pass.cc1087 std::vector<int> input_permutation(args_by_src_.size()); in BuildFunctionDef() local
2579 std::vector<int>* output_permutation, NodeDef* node) { in Run()
Dencapsulate_subgraphs_pass_test.cc806 NodeDef* call_def) { in TEST()
851 NodeDef* call_def) { in TEST()
Dextract_outside_compilation_pass.cc1483 std::unique_ptr<Graph>* graph, std::vector<int>* input_permutation, in operator ()()
/external/tensorflow/tensorflow/compiler/xla/
Dutil.cc90 absl::Span<const int64> input_permutation) { in InversePermutation()