Searched refs:new_perm (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | reorder_reshape_transpose.cc | 81 std::vector<int> new_perm; in ComputeNewPerm() local
82 new_perm.reserve(input_dims.size()); in ComputeNewPerm()
86 new_perm.push_back(intermediate_to_input_indices_map[perm[i]]); in ComputeNewPerm()
92 new_perm.push_back(index); in ComputeNewPerm()
96 CHECK_EQ(new_perm.size(), input_dims.size()); in ComputeNewPerm()
97 return new_perm; in ComputeNewPerm()
216 const std::vector<int> new_perm = in Run() local
218 CHECK_EQ(input_dims.size(), new_perm.size()); in Run()
222 transpose_array.GetMutableBuffer<ArrayDataType::kInt32>().data = new_perm; in Run()
224 static_cast<int>(new_perm.size())}; in Run()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_functor_gpu.cu.cc | 100 TransposePermsVec new_perm; in run() local
102 ReduceTransposeDimensions(in.shape(), perm, &new_perm, &new_dims); in run()
112 if (new_perm[0] == 1 && new_perm[1] == 0) { in run()
122 if (new_perm == TransposePermsVec({0, 2, 1})) { in run()
127 } else if (new_perm == TransposePermsVec({2, 1, 0})) { in run()
|
| D | transpose_util_test.cc | 29 internal::TransposePermsVec new_perm; in TestDimensionReduction() local
31 internal::ReduceTransposeDimensions(shape, perm, &new_perm, &new_dims); in TestDimensionReduction()
33 gtl::ArraySlice<int32> computed_perm(new_perm); in TestDimensionReduction()
|
| D | transpose_functor.h | 80 TransposePermsVec* new_perm, in ReduceTransposeDimensions() argument
85 new_perm->resize(1); in ReduceTransposeDimensions()
86 (*new_perm)[0] = perm[0]; in ReduceTransposeDimensions()
111 new_perm->resize(dim_idx + 1); in ReduceTransposeDimensions()
117 (*new_perm)[dim_idx] = new_perm_idx; in ReduceTransposeDimensions()
|
| /external/selinux/libsepol/src/ |
| D | link.c | 127 perm_datum_t *perm, *new_perm = NULL, *dest_perm; in permission_copy_callback() local
155 new_perm = in permission_copy_callback()
157 if (new_perm == NULL) { in permission_copy_callback()
164 (hashtab_datum_t) new_perm); in permission_copy_callback()
170 new_perm->s.value = dest_class->permissions.nprim + 1; in permission_copy_callback()
171 dest_perm = new_perm; in permission_copy_callback()
204 free(new_perm); in permission_copy_callback()
1265 class_perm_node_t *cur_perm, *new_perm, *tail_perm = NULL; in copy_avrule_list() local
1291 if ((new_perm = (class_perm_node_t *) in copy_avrule_list()
1295 class_perm_node_init(new_perm); in copy_avrule_list()
[all …]
|
| D | expand.c | 195 perm_datum_t *perm, *new_perm; in perm_copy_callback() local
201 new_perm = (perm_datum_t *) malloc(sizeof(perm_datum_t)); in perm_copy_callback()
202 if (!new_perm) { in perm_copy_callback()
205 memset(new_perm, 0, sizeof(perm_datum_t)); in perm_copy_callback()
209 free(new_perm); in perm_copy_callback()
213 new_perm->s.value = perm->s.value; in perm_copy_callback()
216 ret = hashtab_insert(s->table, new_id, (hashtab_datum_t *) new_perm); in perm_copy_callback()
219 free(new_perm); in perm_copy_callback()
2654 class_perm_node_t *cur_perm, *new_perm, *tail_perm; in copy_neverallow() local
2691 new_perm = in copy_neverallow()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1747 new_perm = array_ops.concat([[0], perm + 1], axis=0)
1748 return wrap(array_ops.transpose(t, new_perm), True)
|