Home
last modified time | relevance | path

Searched refs:TransposeParams (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dtranspose_utils_test.cc27 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 …]
Dtranspose_utils.h26 bool IsTranspose2DApplicable(const TransposeParams& params,
34 TransposeParams* params);
43 const RuntimeShape& output_shape, const TransposeParams& params,
46 TransposeParams* non_flatten_params);
Dtranspose_utils.cc20 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()
Dtypes.h1143 struct TransposeParams { struct
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dtranspose.h26 void TransposeImpl(const TransposeParams& params, in TransposeImpl()
73 void Transpose(const TransposeParams& params, in Transpose()
Dlegacy_reference_ops.h969 TransposeParams params; in Transpose()
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose.cc101 TransposeParams params; in Eval()
Dtranspose_test.cc57 TransposeParams params; in RunTestPermutation()
Dtranspose_conv.cc206 TransposeParams transpose_params; in ResizeAndTransposeWeights()
Dbatch_matmul.cc393 TransposeParams params; in TransposeRowsColumnsImpl()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dtranspose.cc101 TransposeParams params; in Eval()
/external/tensorflow/tensorflow/lite/delegates/coreml/
Dcoreml_delegate_kernel.mm47 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/
Dtranspose_test.cc81 TransposeParams params; in ComputeExpectedTransposeResult()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_helpers.cc177 TransposeParams transpose_params; in InitializeWeightsNodes()
Dmatmul_builder.cc175 TransposeParams transpose_params; in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconvolution_op_builder.cc152 TransposeParams params; in TransposeKernelWeights()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h1240 struct TransposeParams {
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h7774 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()
Dlegacy_optimized_ops.h4942 TransposeParams params; in Transpose()