Home
last modified time | relevance | path

Searched refs:GetPermutation (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.h649 std::vector<int> GetPermutation(
Dgeneric_layout_optimizer_transposer.cc258 this->src_to_dst = GetPermutation(this->src_dim_indices, dst_format); in AssignDeviceAndDataFormats()
259 this->dst_to_src = GetPermutation(this->dst_dim_indices, src_format); in AssignDeviceAndDataFormats()
2198 std::vector<int> GetPermutation( in GetPermutation() function
Darithmetic_optimizer.cc1137 TF_RETURN_IF_ERROR(GetPermutation(*node_perm, &node_perm_values)); in TrySimplify()
1148 GetPermutation(*first_transpose_perm, &first_transpose_perm_values)); in TrySimplify()
1185 Status GetPermutation(const NodeDef& node_perm, in GetPermutation() function in tensorflow::grappler::__anon327bfa1e0111::RemoveIdentityTranspose
Dgeneric_layout_optimizer_transposer_test.cc3843 std::vector<int> permutation = GetPermutation(src_dim_indices, dst_format); in TEST()
3861 std::vector<int> permutation = GetPermutation(src_dim_indices, dst_format); in TEST()