Searched defs:get_output_shape (Results 1 – 8 of 8) sorted by relevance
| /external/ComputeLibrary/tests/validation/reference/ |
| D | L2NormalizeLayer.cpp | 42 TensorShape get_output_shape(TensorShape shape, unsigned int axis) in get_output_shape() function
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEGEMMTranspose1xWKernel.cpp | 41 TensorShape get_output_shape(const ITensorInfo *input) in get_output_shape() function
|
| D | NEWeightsReshapeKernel.cpp | 35 TensorShape get_output_shape(const ITensorInfo *input, bool has_bias) in get_output_shape() function
|
| D | NEWinogradConvolutionLayerKernel.cpp | 463 …TransformOutputKernel<T, OutputTileRows, OutputTileCols, KernelRows, KernelCols>::get_output_shape( in get_output_shape() function in arm_compute::NEWinogradLayerTransformOutputKernel
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLPermuteKernel.cpp | 39 TensorShape get_output_shape(const ITensorInfo *input, const PermutationVector &perm) in get_output_shape() function
|
| /external/ComputeLibrary/examples/ |
| D | gc_dc.cpp | 109 …TensorShape get_output_shape(TensorShape in_shape, TensorShape kernel_shape, const PadStrideInfo &… in get_output_shape() function in GCDCExample
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | DirectConvolutionLayerTensorShiftFixture.h | 190 …TensorShape get_output_shape(TensorShape in_shape, TensorShape kernel_shape, const PadStrideInfo &… in get_output_shape() function
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core_test.py | 147 def get_output_shape(input_shape): function
|