| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | convolution_selector.cc | 36 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionAdreno()
49 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionNVidia()
61 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionApple()
79 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolution()
98 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionForWinograd()
108 const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionWithDynamicWeights()
118 const OHWI& weights_shape, const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionBatchedMatMul()
|
| D | fully_connected_selector.cc | 31 BHWC dst_shape = BHWC(batch_size, 1, 1, attr.weights.shape.o); in SelectFullyConnected() local
|
| /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()
27 uint64_t GetConvolutionWinograd4x4To6x6Flops(const BHWC& dst_shape, in GetConvolutionWinograd4x4To6x6Flops()
39 uint64_t GetDepthwiseConvolutionFlops(const BHWC& dst_shape, in GetDepthwiseConvolutionFlops()
46 uint64_t GetFullyConnectedFlops(const BHWC& dst_shape, in GetFullyConnectedFlops()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_generic.cc | 175 const GpuInfo& gpu_info, const BHWC* dst_shape) in ConvGeneric()
202 const BHWC* dst_shape) in ConvGeneric()
213 const GpuInfo& gpu_info, const BHWC* dst_shape) in ConvGeneric()
238 const GpuInfo& gpu_info, const BHWDC* dst_shape) in ConvGeneric()
1136 int GetGroupsCount(const BHWC& dst_shape, const int3& wg_size, in GetGroupsCount()
1148 int GetGroupsCountForLinearWH(const BHWC& dst_shape, const int3& wg_size, in GetGroupsCountForLinearWH()
1160 int GetGroupsCountForLinearWHS(const BHWC& dst_shape, const int3& wg_size, in GetGroupsCountForLinearWHS()
1184 const BHWC& dst_shape) { in GetMaximumPossibleWavesCount()
1193 const BHWC& dst_shape) { in GetRecommendedBlockSize()
1218 const BHWC& dst_shape, const int4& block_size) { in CreateWorkGroupSizeOption()
[all …]
|
| D | conv_metal_simd.cc | 419 int Get2dGroupsCount(const BHWC& dst_shape, const int2 group_size) { in Get2dGroupsCount()
425 int2 GetOptimalGroupSize(const BHWC& dst_shape) { in GetOptimalGroupSize()
455 const OperationDef& definition, const BHWC& dst_shape, in CreateConvolutionMetalSimd()
543 bool IsGoodTaskSizeForAppleConvSimd(const BHWC& dst_shape, in IsGoodTaskSizeForAppleConvSimd()
|
| /external/ComputeLibrary/src/runtime/heuristics/direct_conv/ |
| D | ClDirectConvDefaultConfigValhall.cpp | 81 …const TensorShape dst_shape = misc::shape_calculator::compute_deep_convolution_sh… in configure_G78_f32() local
142 …const TensorShape dst_shape = misc::shape_calculator::compute_deep_convolution_sh… in configure_G78_f16() local
278 …const TensorShape dst_shape = misc::shape_calculator::compute_deep_convolution_sh… in configure_G57_f32() local
338 …const TensorShape dst_shape = misc::shape_calculator::compute_deep_convolution_sh… in configure_G57_f16() local
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClPermuteKernel.cpp | 50 TensorShape dst_shape = src->tensor_shape(); in get_dst_shape() local
71 …const TensorShape dst_shape = misc::shape_calculator::compute_permutation_output_shape(*src, perm); in validate_arguments() local
89 const TensorShape dst_shape = get_dst_shape(src, perm); in configure() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | DepthwiseConvolutionLayer.cpp | 54 …onst SimpleTensor<T> &weights, const SimpleTensor<T> &biases, const TensorShape &dst_shape, const … in depthwise_convolution_fp()
138 …const SimpleTensor<TW> &weights, const SimpleTensor<int32_t> &biases, const TensorShape &dst_shape, in depthwise_convolution_quantized()
238 …onst SimpleTensor<float> &weights, const SimpleTensor<float> &biases, const TensorShape &dst_shape, in depthwise_convolution()
245 … const SimpleTensor<half> &weights, const SimpleTensor<half> &biases, const TensorShape &dst_shape, in depthwise_convolution()
252 … SimpleTensor<uint8_t> &weights, const SimpleTensor<int32_t> &biases, const TensorShape &dst_shape, in depthwise_convolution()
259 …t SimpleTensor<int8_t> &weights, const SimpleTensor<int32_t> &biases, const TensorShape &dst_shape, in depthwise_convolution()
266 …t SimpleTensor<int8_t> &weights, const SimpleTensor<int32_t> &biases, const TensorShape &dst_shape, in depthwise_convolution()
|
| D | ReductionOperation.cpp | 184 …r<OT> compute_reduction_operation(const SimpleTensor<T> &src, const TensorShape &dst_shape, unsign… in compute_reduction_operation()
278 SimpleTensor<OT> reduction_operation(const SimpleTensor<T> &src, const TensorShape &dst_shape, unsi… in reduction_operation()
285 …int8_t> reduction_operation(const SimpleTensor<uint8_t> &src, const TensorShape &dst_shape, unsign… in reduction_operation()
308 …<int8_t> reduction_operation(const SimpleTensor<int8_t> &src, const TensorShape &dst_shape, unsign… in reduction_operation()
|
| D | SliceOperations.cpp | 55 …TensorShape dst_shape = arm_compute::misc::shape_calculator::compute_slice_shape(src_shape, starts… in slice() local
100 …const TensorShape dst_shape = compute_strided_slice_output_shape(src_shape, starts, ends, strides,… in strided_slice() local
|
| D | Transpose.cpp | 41 TensorShape dst_shape{ src.shape().y(), src.shape().x() }; in transpose() local
|
| D | Permute.cpp | 41 TensorShape dst_shape = src.shape(); in permute() local
|
| D | WeightsReshape.cpp | 38 …pe(const SimpleTensor<T> &src, const SimpleTensor<T> &biases, const TensorShape &dst_shape, const … in weights_reshape()
|
| D | Col2Im.cpp | 38 SimpleTensor<T> col2im(const SimpleTensor<T> &src, const TensorShape &dst_shape, unsigned int num_g… in col2im()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | global_pooling_to_reduce_op.cc | 35 const BHWC& dst_shape) { in IsGlobalPooling()
43 const BHWC& src_shape, const BHWC& dst_shape) { in IsGlobalAveragePooling()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | BatchNormalizationLayerFusionFixture.h | 49 …(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorShape dst_shape, PadStride… in setup()
83 …(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorShape dst_shape, PadStride… in compute_target()
159 …(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorShape dst_shape, PadStride… in compute_reference()
|
| D | ConcatenateLayerFixture.h | 113 … const TensorShape dst_shape = misc::shape_calculator::calculate_concatenate_shape(src_ptrs, axis); in compute_target() local
163 … const TensorShape dst_shape = misc::shape_calculator::calculate_concatenate_shape(src_ptrs, axis); in compute_reference() local
|
| D | FullyConnectedLayerFixture.h | 353 …up(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape dst_shape, in setup()
474 …up(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape dst_shape, in setup()
487 …up(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape dst_shape, in setup()
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuConcatenate.cpp | 54 …TensorShape dst_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(srcs_vect… in configure() local
140 …TensorShape dst_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(srcs_vect… in validate() local
|
| /external/ComputeLibrary/src/runtime/heuristics/indirect_conv/ |
| D | ClIndirectConvDefaultConfigValhall.cpp | 66 …const TensorShape dst_shape = misc::shape_calculator::compute_deep_convolution_sh… in configure_G77_f32() local
110 …const TensorShape dst_shape = misc::shape_calculator::compute_deep_convolution_sh… in configure_G77_f16() local
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClConcatenate.cpp | 54 …TensorShape dst_shape = arm_compute::misc::shape_calculator::calculate_concat… in configure() local
212 …TensorShape dst_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(src_vecto… in validate() local
|
| /external/ComputeLibrary/src/cpu/kernels/internal/ |
| D | CpuDepthwiseConv2dAssemblyWrapperKernel.cpp | 217 const TensorShape dst_shape = compute_depthwise_convolution_shape(*src, *weights, info); in configure() local
297 …const TensorShape dst_shape = misc::shape_calculator::compute_depthwise_convolution_shape(*src, *w… in validate() local
324 const auto dst_shape = dst->info()->tensor_shape(); in run_op() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_swish_op.cc | 46 TensorShape dst_shape = src_shape; in Compute_Scalar() local
|
| D | mkl_fused_mish_op.cc | 45 TensorShape dst_shape = src_shape; in Compute_Scalar() local
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuPermuteKernel.cpp | 108 …const TensorShape dst_shape = misc::shape_calculator::compute_permutation_output_shape(*src, perm); in validate_arguments() local
246 …const TensorShape dst_shape = misc::shape_calculator::compute_permutation_output_shape(*src, perm); in configure() local
|