Searched defs:original_input_shape (Results 1 – 8 of 8) sorted by relevance
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLConvertFullyConnectedWeights.cpp | 32 …ghts::configure(const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, in configure()
38 …ompile_context, const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, in configure()
46 …idate(const ITensorInfo *input, const ITensorInfo *output, const TensorShape &original_input_shape, in validate()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLConvertFullyConnectedWeightsKernel.cpp | 42 …rnel::configure(const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, in configure()
48 …ompile_context, const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, in configure()
91 …idate(const ITensorInfo *input, const ITensorInfo *output, const TensorShape &original_input_shape, in validate()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEConvertFullyConnectedWeights.cpp | 37 …dWeights::configure(const ITensor *input, ITensor *output, const TensorShape &original_input_shape, in configure()
44 …idate(const ITensorInfo *input, const ITensorInfo *output, const TensorShape &original_input_shape, in validate()
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLConvertFullyConnectedWeights.h | 111 …void configure(const ICLTensor *input, const TensorShape &original_input_shape, DataLayout data_la… in configure()
122 …CompileContext &compile_context, const ICLTensor *input, const TensorShape &original_input_shape, … in configure()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEConvertFullyConnectedWeightsKernel.cpp | 38 …tsKernel::configure(const ITensor *input, ITensor *output, const TensorShape &original_input_shape, in configure()
68 …idate(const ITensorInfo *input, const ITensorInfo *output, const TensorShape &original_input_shape, in validate()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NEConvertFullyConnectedWeights.h | 109 …void configure(const ITensor *input, const TensorShape &original_input_shape, DataLayout data_layo… in configure()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | ConvertFullyConnectedWeights.cpp | 35 …vert_fully_connected_weights(const SimpleTensor<T> &src, const TensorShape &original_input_shape, … in convert_fully_connected_weights()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 400 const ArrayRef<int64_t> &original_input_shape = in RewriteNewAxisMask() local
|