Searched refs:TransposeParams (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils_test.cc | 27 TransposeParams params; in TEST()
45 TransposeParams params; in TEST()
65 TransposeParams params; in TEST()
84 TransposeParams params; in TEST()
106 TransposeParams params; in TEST()
128 TransposeParams params; in TEST()
148 TransposeParams params; in TEST()
168 TransposeParams params; in TEST()
192 TransposeParams params; in TEST()
215 TransposeParams params; in TEST()
[all …]
|
| D | transpose_utils.h | 26 bool IsTranspose2DApplicable(const TransposeParams& params,
34 TransposeParams* params);
43 const RuntimeShape& output_shape, const TransposeParams& params,
46 TransposeParams* non_flatten_params);
|
| D | transpose_utils.cc | 20 bool IsTranspose2DApplicable(const TransposeParams& params, in IsTranspose2DApplicable()
55 TransposeParams* params) { in RemoveOneSizeDimensions()
93 TransposeParams new_params; in RemoveOneSizeDimensions()
121 const RuntimeShape& output_shape, const TransposeParams& params, in Flatten()
124 TransposeParams* non_flatten_params) { in Flatten()
|
| D | types.h | 1143 struct TransposeParams { struct
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | transpose.h | 26 void TransposeImpl(const TransposeParams& params, in TransposeImpl()
73 void Transpose(const TransposeParams& params, in Transpose()
|
| D | legacy_reference_ops.h | 969 TransposeParams params; in Transpose()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose.cc | 101 TransposeParams params; in Eval()
|
| D | transpose_test.cc | 57 TransposeParams params; in RunTestPermutation()
|
| D | batch_matmul.cc | 393 TransposeParams params; in TransposeRowsColumnsImpl()
|
| D | transpose_conv.cc | 206 TransposeParams transpose_params; in ResizeAndTransposeWeights()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | transpose.cc | 101 TransposeParams params; in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 47 TransposeParams params = {/*perm_count=*/4, /*perm=*/{0, 3, 1, 2}};
56 TransposeParams params = {/*perm_count=*/4, /*perm=*/{0, 2, 3, 1}};
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_test.cc | 81 TransposeParams params; in ComputeExpectedTransposeResult()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_helpers.cc | 177 TransposeParams transpose_params; in InitializeWeightsNodes()
|
| D | matmul_builder.cc | 175 TransposeParams transpose_params; in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | convolution_op_builder.cc | 152 TransposeParams params; in TransposeKernelWeights()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 1240 struct TransposeParams {
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 7774 inline void Transpose3D(const TransposeParams& params, in Transpose3D()
7824 void TransposeImpl(const TransposeParams& params, in TransposeImpl()
7859 void Transpose(const TransposeParams& unshrinked_params, in Transpose()
7871 TransposeParams shrinked_params = unshrinked_params; in Transpose()
7898 TransposeParams non_flatten_params; in Transpose()
|
| D | legacy_optimized_ops.h | 4942 TransposeParams params; in Transpose()
|