Home
last modified time | relevance | path

Searched refs:IsIdentityPermutation (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dscatter_simplifier.cc184 if (IsIdentityPermutation(operand_permutation)) { in ExpandInstruction()
216 !IsIdentityPermutation(dims.scatter_dims_to_operand_dims()); in InstructionMatchesPattern()
Dgather_scatter_utils.cc67 if (IsIdentityPermutation(permutation)) { in MaybeTranspose()
Dgather_simplifier.cc137 !IsIdentityPermutation(dims.start_index_map()) || in InstructionMatchesPattern()
/external/tensorflow/tensorflow/compiler/xla/
Dpermutation_util.cc55 bool IsIdentityPermutation(absl::Span<const int64_t> permutation) { in IsIdentityPermutation() function
Dpermutation_util.h81 bool IsIdentityPermutation(absl::Span<const int64_t> permutation);
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1227 if (IsIdentityPermutation(node_perm_values)) { in TrySimplify()
1277 bool IsIdentityPermutation(const std::vector<int64_t>& perm) { in IsIdentityPermutation() function in tensorflow::grappler::__anon2c8062030111::RemoveIdentityTranspose
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1190 XlaOp transposed = IsIdentityPermutation(dimensions) in Reshape()