Searched refs:NETranspose (Results 1 – 16 of 16) sorted by relevance
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NETranspose.h | 40 class NETranspose : public IFunction
44 NETranspose();
46 ~NETranspose();
48 NETranspose(const NETranspose &) = delete;
50 NETranspose(NETranspose &&) = default;
52 NETranspose &operator=(const NETranspose &) = delete;
54 NETranspose &operator=(NETranspose &&) = default;
|
| D | NEQLSTMLayer.h | 263 …NETranspose _transpose_input_to_forget_weight…
264 … NETranspose _transpose_input_to_cell_weights;
265 …NETranspose _transpose_input_to_output_weight…
266 …NETranspose _transpose_input_to_input_weights;
267 …NETranspose _transpose_recurrent_to_forget_we…
268 …NETranspose _transpose_recurrent_to_cell_weig…
269 …NETranspose _transpose_recurrent_to_output_we…
270 …NETranspose _transpose_recurrent_to_input_wei…
271 NETranspose _transpose_projection_weights;
|
| D | NEFullyConnectedLayer.h | 74 NETranspose _func{};
|
| D | NELSTMLayerQuantized.h | 152 NETranspose _transpose_weights;
|
| D | NELSTMLayer.h | 179 NETranspose _transpose_cell_state;
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NETranspose.cpp | 32 struct NETranspose::Impl
39 NETranspose::NETranspose() in NETranspose() function in arm_compute::NETranspose
44 NETranspose::~NETranspose() = default;
46 void NETranspose::configure(const ITensor *input, ITensor *output) in configure()
57 Status NETranspose::validate(const ITensorInfo *input, const ITensorInfo *output) in validate()
64 void NETranspose::run() in run()
|
| D | NEQLSTMLayer.cpp | 802 …ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(input_to_cell_weights, &input_weights_transposed… in validate()
803 …ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(input_to_output_weights, &input_to_output_weight… in validate()
804 …ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(recurrent_to_forget_weights, &recurrent_to_forge… in validate()
805 …ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(recurrent_to_cell_weights, &recurrent_to_cell_we… in validate()
806 …ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(recurrent_to_output_weights, &recurrent_to_outpu… in validate()
813 …ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(lstm_params.input_to_input_weights(), &input_to_… in validate()
814 …ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(lstm_params.recurrent_to_input_weights(), &recur… in validate()
819 …ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(lstm_params.projection_weights(), &projection_we… in validate()
|
| D | NELSTMLayerQuantized.cpp | 324 ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(&weights, &weights_transposed)); in validate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_matmul_acl.h | 31 arm_compute::NETranspose trans_lhs;
32 arm_compute::NETranspose trans_rhs;
|
| D | runtime_matmul_acl.cc | 85 auto acl_trans_lhs_st = arm_compute::NETranspose::validate( in MatMulF32()
93 auto acl_trans_rhs_st = arm_compute::NETranspose::validate( in MatMulF32()
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | Transpose.cpp | 63 Status status = NETranspose::validate(&a_info.clone()->set_is_resizable(false),
71 using NETransposeFixture = TransposeValidationFixture<Tensor, Accessor, NETranspose, T>;
|
| /external/ComputeLibrary/docs/user_guide/ |
| D | release_version_and_change_log.dox | 1553 - NETransposeKernel / @ref NETranspose
|
| D | operator_list.dox | 3077 <td>NETranspose
|
| /external/ComputeLibrary/src/ |
| D | CMakeLists.txt | 947 runtime/NEON/functions/NETranspose.cpp
|
| D | BUILD.bazel | 955 "runtime/NEON/functions/NETranspose.cpp",
|
| /external/ComputeLibrary/ |
| D | Android.bp | 868 "src/runtime/NEON/functions/NETranspose.cpp",
|