Home
last modified time | relevance | path

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

12

/external/ComputeLibrary/tests/validation/fixtures/
DFullyConnectedLayerFixture.h58 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
103 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
193 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
221 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
235 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
DFFTFixture.h121 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
148 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
198 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
228 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
DConvolutionLayerFixture.h71 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
156 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
213 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
264 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
278 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
291 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
DRNNLayerFixture.h46 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape recurrent_weights_shape… in setup()
61 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
111 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
DDirectConvolutionLayerFixture.h76 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
128 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
175 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
228 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
241 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
DDirectConvolutionLayerTensorShiftFixture.h69 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
113 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
164 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
234 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
247 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
DConvertFullyConnectedWeightsFixture.h80 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
110 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
DWinogradConvolutionLayerFixture.h59 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
88 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_sha… in compute_target()
129 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
163 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
197 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_sha… in compute_target()
246 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
DDepthwiseConvolutionLayerFixture.h118 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape biases_s… in compute_target()
167 …SimpleTensor<T> compute_reference(const TensorShape &in_shape, const TensorShape &weights_shape, c… in compute_reference()
246 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape biases_s… in compute_target()
292 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
361 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape biases_s… in compute_target()
413 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
DDeconvolutionLayerFixture.h53 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
126 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape bi… in compute_target()
185 …Tensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape, cons… in compute_reference()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter_test_util.cc110 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX4Test() local
144 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX4UnalignedTest() local
177 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX2Test() local
210 auto weights_shape = OHWI(kDstChannels, 3, 4, kSrcChannels); in ConverterToConvWeightsOutX2Test() local
244 auto weights_shape = OHWI(kDstChannels, 4, 4, kSrcChannels); in ConverterToConvTransposedWeights4x4Test() local
279 auto weights_shape = OHWI(dst_channels, 1, 1, src_channels); in ConverterToConvWeights4xTexturesTest() local
Dconv_buffer_1x1.cc377 const BHWC& weights_shape, in IsConvBuffer1x1Supported()
448 const Convolution2DAttributes& attr, const BHWC& weights_shape, in CreateConvBuffer1x1DynamicWeights()
/external/tensorflow/tensorflow/python/ops/
Dweights_broadcast_ops.py34 def _has_valid_dims(weights_shape, values_shape): argument
48 weights_rank, weights_shape, values_rank, values_shape): argument
/external/ComputeLibrary/tests/validation/fixtures/UNIT/
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
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h33 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
79 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4Impl()
149 const RuntimeShape& weights_shape; member
167 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc59 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionDynamicWeightsAdreno()
117 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionDynamicWeightsMali()
184 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionWithDynamicWeights()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/
Dfully_connected.h28 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dcreate_im2col_arrays.cc40 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator() local
Dconvert_pure_conv_to_depthwise.cc98 const auto& weights_shape = weights_array.shape(); in Run() local
Densure_bias_vectors.cc31 const auto& weights_shape = model.GetArray(weights_name).shape(); in GetOutputDepthFromWeights() local
Dshuffle_fc_weights.cc91 const Shape& weights_shape = weights_array.shape(); in Run() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc203 BHWC weights_shape(second_shape.c, 1, 1, second_shape.w); in GPUOperationFromNode() local
339 auto weights_shape = inputs[1]->tensor.shape; in GPUOperationFromNode() local
388 OHWI weights_shape = in GPUOperationFromNode() local
/external/ComputeLibrary/examples/
Dgc_dc.cpp57 const TensorShape weights_shape(kernel_size, kernel_size, src_shape.z(), num_kernels); in do_setup() local
/external/ComputeLibrary/tests/validation/NEON/
DDirectConvolutionLayer.cpp207 TensorShape weights_shape(kernel_size, kernel_size, input_shape.z(), num_kernels); variable
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h28 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
166 const uint8_t* input_data, const RuntimeShape& weights_shape, in ShuffledFullyConnected()

12