| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | flops_util.cc | 21 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()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | FullyConnectedLayerFixture.h | 124 …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()
|
| D | DirectConvolutionLayerFixture.h | 77 …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()
|
| D | RNNLayerFixture.h | 46 …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()
|
| D | ConvertFullyConnectedWeightsFixture.h | 85 …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()
|
| D | FFTFixture.h | 186 …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()
|
| D | ConvolutionLayerFixture.h | 366 …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()
|
| D | DepthwiseConvolutionLayerFixture.h | 96 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
|
| D | DeconvolutionLayerFixture.h | 53 …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()
|
| D | DirectConvolution3DFixture.h | 85 …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()
|
| D | WinogradConvolutionLayerFixture.h | 59 …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()
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | ConvolutionLayer.cpp | 97 …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
|
| /external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/ |
| D | DirectConv2dFixture.h | 92 …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()
|
| D | DepthwiseConv2dFixture.h | 78 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()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter_test_util.cc | 135 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
|
| D | conv_weights_converter.cc | 211 const auto& weights_shape = GetWeightsSize(); in BindArguments() local
231 const auto& weights_shape = GetWeightsSize(); in GetGridSize() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | weights_broadcast_ops.py | 30 def _has_valid_dims(weights_shape, values_shape): argument
44 weights_rank, weights_shape, values_rank, values_shape): argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| D | fully_connected.h | 36 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()
|
| /external/ComputeLibrary/tests/validation/fixtures/UNIT/ |
| D | DynamicTensorFixture.h | 255 …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/ComputeLibrary/tests/validation/NEON/ |
| D | DirectConvolutionLayer.cpp | 148 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
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | convolution_selector.cc | 107 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionWithDynamicWeights()
118 const OHWI& weights_shape, const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionBatchedMatMul()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/ |
| D | fully_connected.h | 28 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | create_im2col_arrays.cc | 40 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator() local
|
| D | convert_pure_conv_to_depthwise.cc | 98 const auto& weights_shape = weights_array.shape(); in Run() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | operation_selector.cc | 160 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
|