Searched refs:transpose_perm (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | merge_reshape_into_preceding_transpose.cc | 169 const auto& transpose_perm = transpose_op->perm; in Run() local
171 merged_perm[i] = transpose_perm[merged_perm[i]]; in Run()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_kronecker.py | 57 transpose_perm = array_ops.concat(
60 transpose_perm = array_ops.concat(
62 return array_ops.transpose(x, transpose_perm)
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 632 const std::string transpose_perm = in ConvertFullyConnectedOperator() local
634 CreateIntTensorConst(transpose_perm, {1, 0}, {2}, tensorflow_graph); in ConvertFullyConnectedOperator()
639 *transpose_op->add_input() = transpose_perm; in ConvertFullyConnectedOperator()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5805 nvinfer1::Permutation transpose_perm; in ConvertDepthSpaceShuffle() local
5819 transpose_perm = {2, 3, 0, 4, 1}; in ConvertDepthSpaceShuffle()
5835 transpose_perm = {2, 4, 0, 1, 3}; in ConvertDepthSpaceShuffle()
5851 first_shuffle->setSecondTranspose(transpose_perm); in ConvertDepthSpaceShuffle()
|