Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dpermutation_util.cpp11 c10::ArrayRef<int64_t> input_permutation) { in InversePermutation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dpermutation_util.cc35 absl::Span<const int64_t> input_permutation) { in InversePermutation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A 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()
H A Dencapsulate_xla_computations_pass.cc116 std::vector<int>* input_permutation, in RewriteSubgraph()
H A Dencapsulate_subgraphs_pass_test.cc799 NodeDef* call_def) { in TEST()
844 NodeDef* call_def) { in TEST()
H A Dextract_outside_compilation_pass.cc2205 std::unique_ptr<Graph>* graph, std::vector<int>* input_permutation, in operator ()()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc79 std::vector<int>* input_permutation, in RewriteSubgraph()