Home
last modified time | relevance | path

Searched defs:weights_shape (Results 1 – 25 of 52) sorted by relevance

123

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DFullyConnectedLayerFixture.h124 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
223 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
252 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
266 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
308 void fill_transposed_weights(TensorType &weights, TensorShape weights_shape, int seed) in fill_transposed_weights()
353 …void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape d… in setup()
474 …void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape d… in setup()
487 …void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape d… in setup()
H A DDirectConvolutionLayerFixture.h77 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
149 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
206 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
260 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
273 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
H A DConvolutionLayerFixture.h366 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
394 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
407 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
518 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
564 …void compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_sh… in compute_target()
593 …void compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, const Ten… in compute_reference()
H A DRNNLayerFixture.h46 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape recurrent_weights_shape… in setup()
64 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
114 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
H A DFFTFixture.h186 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
244 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
275 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
H A DDepthwiseConvolutionLayerFixture.h96 TensorShape weights_shape = _weights_shape; in configure_target() local
332 TensorShape weights_shape = _weights_shape; in configure_target() local
488 TensorShape weights_shape = _weights_shape; in configure_target() local
H A DDeconvolutionLayerFixture.h53 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
153 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape bi… in compute_target()
211 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
H A DConvertFullyConnectedWeightsFixture.h85 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
115 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
H A DWinogradConvolutionLayerFixture.h59 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
109 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_sha… in compute_target()
166 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
H A DDirectConvolution3DFixture.h85 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
130 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dflops_util.cc21 uint64_t GetConvolutionFlops(const BHWC& dst_shape, const OHWI& weights_shape) { in GetConvolutionFlops()
28 const OHWI& weights_shape) { in GetConvolutionWinograd4x4To6x6Flops()
33 const OHWI& weights_shape) { in GetConvolutionTransposedFlops()
40 const OHWI& weights_shape) { in GetDepthwiseConvolutionFlops()
47 const OHWI& weights_shape) { in GetFullyConnectedFlops()
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DConvolutionLayer.cpp97 …t_op_list_valid_in_gemmconv(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in is_post_op_list_valid_in_gemmconv()
296 const auto weights_shape = TensorShape(16U, 1U, 1U, 24U); in TEST_SUITE() local
323 const auto weights_shape = TensorShape(1U, 1U, 16U, 24U); in TEST_CASE() local
344 const auto weights_shape = TensorShape(16U, 1U, 1U, 24U); in TEST_CASE() local
365 const auto weights_shape = TensorShape(16U, 3U, 3U, 24U); in TEST_CASE() local
386 const auto weights_shape = TensorShape(16U, 1U, 1U, 24U); in TEST_CASE() local
409 const auto weights_shape = TensorShape(16U, 1U, 1U, 24U); in TEST_SUITE() local
423 const auto weights_shape = TensorShape(16U, 1U, 1U, 24U); in TEST_CASE() local
H A DDirectConvolutionLayer.cpp101 const auto weights_shape = TensorShape(3U, 3U, 2U, 4U); in TEST_SUITE() local
144 auto weights_shape = TensorShape(5U, 7U, 3U, 4U); // non-square kernel in TEST_CASE() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
H A DDirectConv2dFixture.h92 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
109 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
169 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
207 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape output_shape, TensorSha… in setup()
248 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
320 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
H A DDepthwiseConv2dFixture.h78 TensorShape weights_shape = TensorShape(kernel_size.width, kernel_size.height); in setup() local
119 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
182 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_weights_converter_test_util.cc135 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX4Test() local
169 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX4UnalignedTest() local
202 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX2Test() local
235 auto weights_shape = OHWI(kDstChannels, 3, 4, kSrcChannels); in ConverterToConvWeightsOutX2Test() local
269 auto weights_shape = OHWI(kDstChannels, 4, 4, kSrcChannels); in ConverterToConvTransposedWeights4x4Test() local
304 auto weights_shape = OHWI(dst_channels, 1, 1, src_channels); in ConverterToConvWeights4xTexturesTest() local
H A Dconv_weights_converter.cc211 const auto& weights_shape = GetWeightsSize(); in BindArguments() local
231 const auto& weights_shape = GetWeightsSize(); in GetGridSize() local
H A Dconv_generic.cc201 const BHWC& weights_shape, const GpuInfo& gpu_info, in ConvGeneric()
1876 const Convolution2DAttributes& attr, const BHWC& weights_shape, in GuessBestParams()
1906 const OHWI& weights_shape, const BHWC* dst_shape) { in GuessBestParamsPointwise()
1941 const BHWC& weights_shape, in CreateConvGenericDynamicWeights()
1951 const OHWI& weights_shape, in CreateConvGenericBatchedMatMul()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
H A Dfully_connected.h36 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
82 const RuntimeShape& weights_shape, const int8_t* weights_data, in FullyConnectedSparseWeight1x16Impl()
118 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4Impl()
188 const RuntimeShape& weights_shape; member
202 const RuntimeShape& weights_shape, const int8_t* weights_data, in FullyConnectedSparseWeight1x16()
226 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/UNIT/
H A DDynamicTensorFixture.h255 …void setup(std::vector<TensorShape> input_shapes, TensorShape weights_shape, TensorShape bias_shap… in setup()
296 …TensorType run_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, … in run_target()
345 …SimpleTensor<T> run_reference(TensorShape input_shape, TensorShape weights_shape, TensorShape bias… in run_reference()
434 TensorShape weights_shape = TensorShape(3U, 3U, input_shape[2], input_shape[2]); in run() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Doperation_selector.cc160 const BHWC& src_shape, const OHWI& weights_shape, in AddDynamicConv()
330 const OHWI weights_shape(second_shape.c, second_shape.b, second_shape.h, in GPUOperationFromNodePart0() local
487 const OHWI weights_shape = in GPUOperationFromNodePart0() local
515 const OHWI weights_shape = in GPUOperationFromNodePart0() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dweights_broadcast_ops.py30 def _has_valid_dims(weights_shape, values_shape): argument
44 weights_rank, weights_shape, values_rank, values_shape): argument
/aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/
H A DDirectConvolutionLayer.cpp148 const auto weights_shape = TensorShape(3U, 3U, 2U, 4U); in TEST_SUITE() local
302 TensorShape weights_shape(kernel_size, kernel_size, input_shape.z(), num_kernels); variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
H A Dconvolution_selector.cc107 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionWithDynamicWeights()
118 const OHWI& weights_shape, const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionBatchedMatMul()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlstm_cell.h33 const float* prev_activ_data, const RuntimeShape& weights_shape, in LstmCell()
233 const RuntimeShape& weights_shape, in LstmCell()

123