Home
last modified time | relevance | path

Searched defs:permutation (Results 1 – 25 of 111) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_layout_helper.cc21 SmallVector<int64_t, 4> ReversePermutation(ArrayRef<int64_t> permutation) { in ReversePermutation()
42 ArrayAttr ShuffleArrayAttr(ArrayAttr attr, ArrayRef<int64_t> permutation, in ShuffleArrayAttr()
62 Type ShuffleRankedTensorType(Type type, ArrayRef<int64_t> permutation) { in ShuffleRankedTensorType()
/external/tensorflow/tensorflow/compiler/xla/
Dpermutation_util.h41 const Container& input, absl::Span<const int64_t> permutation) { in Permute()
60 const Container& input, absl::Span<const int64_t> permutation) { in PermuteInverse()
Dpermutation_util.cc23 bool IsPermutation(absl::Span<const int64_t> permutation) { in IsPermutation()
55 bool IsIdentityPermutation(absl::Span<const int64_t> permutation) { in IsIdentityPermutation()
/external/mesa3d/src/intel/compiler/
Dbrw_fs_bank_conflicts.cpp753 struct permutation { struct
754 permutation() : v(NULL), size(0) {} in permutation() function
756 permutation(unsigned n) : in permutation() argument
759 permutation(const permutation &p) : in permutation() function
765 ~permutation() in ~permutation()
771 operator=(permutation p) in operator =()
778 unsigned *v;
779 unsigned size;
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Ddata_format.cc34 std::vector<int64_t> permutation; in Contract() local
70 std::vector<int64_t> permutation; in Expand() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlayout_optimization.cc46 const SmallVector<int64_t, 4>& permutation, in ReuseExistingTranspose()
141 auto perm_attr = [&](Permutation permutation) -> DenseIntElementsAttr { in runOnOperation()
213 SmallVector<int64_t, 4> permutation; in MoveTransposeBefore() local
279 mlir::ShapedType type, ArrayRef<int64_t> permutation) { in ReversePermuteShapedType()
359 SmallVector<int64_t, 8> permutation; in MoveTransposeAfter() local
367 SmallVector<int64_t, 8> permutation; in MoveTransposeAfter() local
/external/swiftshader/third_party/SPIRV-Tools/source/fuzz/
Dtransformation_permute_phi_operands.cpp29 uint32_t result_id, const std::vector<uint32_t>& permutation) { in TransformationPermutePhiOperands()
55 std::vector<uint32_t> permutation(message_.permutation().begin(), in IsApplicable() local
Dtransformation_permute_function_parameters.cpp32 const std::vector<uint32_t>& permutation) { in TransformationPermuteFunctionParameters()
55 std::vector<uint32_t> permutation(message_.permutation().begin(), in IsApplicable() local
Dfuzzer_pass_permute_function_parameters.cpp59 std::vector<uint32_t> permutation(arg_size); in Apply() local
Dfuzzer_pass_permute_phi_operands.cpp54 std::vector<uint32_t> permutation(inst.NumInOperands() / 2); in Apply() local
/external/deqp-deps/SPIRV-Tools/source/fuzz/
Dtransformation_permute_phi_operands.cpp29 uint32_t result_id, const std::vector<uint32_t>& permutation) { in TransformationPermutePhiOperands()
55 std::vector<uint32_t> permutation(message_.permutation().begin(), in IsApplicable() local
Dtransformation_permute_function_parameters.cpp32 const std::vector<uint32_t>& permutation) { in TransformationPermuteFunctionParameters()
55 std::vector<uint32_t> permutation(message_.permutation().begin(), in IsApplicable() local
Dfuzzer_pass_permute_function_parameters.cpp59 std::vector<uint32_t> permutation(arg_size); in Apply() local
Dfuzzer_pass_permute_phi_operands.cpp54 std::vector<uint32_t> permutation(inst.NumInOperands() / 2); in Apply() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/fuzz/
Dtransformation_permute_phi_operands.cpp29 uint32_t result_id, const std::vector<uint32_t>& permutation) { in TransformationPermutePhiOperands()
55 std::vector<uint32_t> permutation(message_.permutation().begin(), in IsApplicable() local
Dtransformation_permute_function_parameters.cpp32 const std::vector<uint32_t>& permutation) { in TransformationPermuteFunctionParameters()
55 std::vector<uint32_t> permutation(message_.permutation().begin(), in IsApplicable() local
Dfuzzer_pass_permute_function_parameters.cpp59 std::vector<uint32_t> permutation(arg_size); in Apply() local
Dfuzzer_pass_permute_phi_operands.cpp54 std::vector<uint32_t> permutation(inst.NumInOperands() / 2); in Apply() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_scatter_utils.cc66 HloInstruction* operand, absl::Span<const int64_t> permutation) { in MaybeTranspose()
88 std::vector<int64_t> permutation; in MoveDimensionToEnd() local
/external/tensorflow/tensorflow/core/common_runtime/
Dpermuter_test.cc51 const std::vector<int>& permutation, DataType dtype, in Init()
87 std::vector<int> permutation(num_workers * num_devices); in RunTest() local
142 DeviceInstance(int rank, std::vector<int> permutation, DataType dtype, in DeviceInstance()
/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
Dsparse_tensors_map_ops_test.py48 def _SparseTensorValue_5x6(self, permutation): argument
59 def _SparseTensorValue_3x4(self, permutation): argument
/external/rust/crates/nom/src/branch/
Dmod.rs58 fn permutation(&mut self, input: I) -> IResult<I, O, E>; in permutation() method
106 pub fn permutation<I: Clone, O, E: ParseError<I>, List: Permutation<I, O, E>>( in permutation() function
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dlu_decomposition.h54 XlaOp permutation; member
/external/icing/icing/testing/
Drandom-string.cc25 for (int permutation = 0; permutation < num_terms; ++permutation) { in GenerateUniqueTerms() local
/external/tensorflow/tensorflow/core/kernels/
Dtranspose_op.cc97 std::vector<int32>* permutation) { in PermutationHelper()
139 std::vector<int32> permutation; in Compute() local

12345