Searched defs:original_input_shape (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLConvertFullyConnectedWeights.cpp | 49 …ghts::configure(const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, in configure()
55 …ompile_context, const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, in configure()
66 …idate(const ITensorInfo *input, const ITensorInfo *output, const TensorShape &original_input_shape, in validate()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| H A D | CLConvertFullyConnectedWeights.h | 140 …void configure(const ICLTensor *input, const TensorShape &original_input_shape, DataLayout data_la… in configure()
151 …CompileContext &compile_context, const ICLTensor *input, const TensorShape &original_input_shape, … in configure()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEConvertFullyConnectedWeights.cpp | 43 …dWeights::configure(const ITensor *input, ITensor *output, const TensorShape &original_input_shape, in configure()
54 …idate(const ITensorInfo *input, const ITensorInfo *output, const TensorShape &original_input_shape, in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuConvertFullyConnectedWeightsKernel.cpp | 37 …ernel::configure(const ITensorInfo *src, ITensorInfo *dst, const TensorShape &original_input_shape, in configure()
65 …::validate(const ITensorInfo *src, const ITensorInfo *dst, const TensorShape &original_input_shape, in validate()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | ConvertFullyConnectedWeights.cpp | 35 …vert_fully_connected_weights(const SimpleTensor<T> &src, const TensorShape &original_input_shape, … in convert_fully_connected_weights()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | prepare_tf.cc | 442 const ArrayRef<int64_t> &original_input_shape = in RewriteNewAxisMask() local
|