| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | convolution_selector.cc | 37 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionAdreno()
50 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionWinogradAdreno()
60 const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionDynamicWeightsAdreno()
70 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionNVidia()
89 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionMali()
103 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionWinogradMali()
118 const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionDynamicWeightsMali()
138 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolution()
160 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionForWinograd()
185 const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionWithDynamicWeights()
|
| D | fully_connected_selector.cc | 31 BHWC dst_shape = BHWC(batch_size, 1, 1, attr.weights.shape.o); in SelectFullyConnectedGeneric() local
44 BHWC dst_shape = BHWC(batch_size, 1, 1, attr.weights.shape.o); in SelectFullyConnectedAdreno() local
73 BHWC dst_shape = BHWC(batch_size, 1, 1, attr.weights.shape.o); in SelectFullyConnectedMali() local
89 BHWC dst_shape = BHWC(batch_size, 1, 1, attr.weights.shape.o); in SelectFullyConnected() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_powervr.cc | 172 const GpuInfo& gpu_info, const BHWC* dst_shape) in ConvPowerVR()
183 const BHWC* dst_shape) in ConvPowerVR()
194 const GpuInfo& gpu_info, const BHWC* dst_shape) in ConvPowerVR()
219 const GpuInfo& gpu_info, const BHWDC* dst_shape) in ConvPowerVR()
1061 bool different_weights_for_height, const BHWC* dst_shape) { in GuessBestParams()
1341 const Convolution2DAttributes& attr, const BHWC* dst_shape) { in GuessBestParams()
1358 const Convolution3DAttributes& attr, const BHWDC* dst_shape) { in GuessBestParams()
1394 const BHWC* dst_shape) { in GuessBestParams()
1409 const FullyConnectedAttributes& attr, const BHWC* dst_shape) { in GuessBestParams()
1423 const Convolution2DAttributes& attr, const BHWC* dst_shape) { in GuessBestParamsWinograd()
[all …]
|
| D | conv_metal.cc | 620 int GetGroupsCount(const BHWC& dst_shape, const int3& wg_size, in GetGroupsCount()
632 int GetGroupsCountForLinearWH(const BHWC& dst_shape, const int3& wg_size, in GetGroupsCountForLinearWH()
644 int GetGroupsCountForLinearWHS(const BHWC& dst_shape, const int3& wg_size, in GetGroupsCountForLinearWHS()
668 const BHWC& dst_shape) { in GetMaximumPossibleWavesCount()
677 const BHWC& dst_shape) { in GetRecommendedBlockSize()
693 const BHWC& dst_shape) { in GetConvParamsForA7A8()
773 const BHWC& dst_shape) { in GetConvParamsForA9AndHigher()
856 const BHWC& dst_shape) { in GetConvParamsForIntel()
899 const BHWC& dst_shape) { in GetConvParamsForAMD()
924 const BHWC& dst_shape) { in GetConvParams()
[all …]
|
| D | conv_buffer_1x1.cc | 405 const BHWC* dst_shape) { in CreateConvBuffer1x1DynamicWeights()
|
| /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/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | cl_test.cc | 73 auto dst_shape = dst_sizes[i]; in ExecuteGPUOperation() local
121 auto dst_shape = dst_sizes[i]; in ExecuteGPUOperation() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | test_util.cc | 83 auto dst_shape = dst_sizes[i]; in ExecuteGPUOperation() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | operation_selector.cc | 45 const BHWC& dst_shape) { in IsRecommendedForWinograd4x4To6x6()
196 auto dst_shape = outputs[0]->tensor.shape; in GPUOperationFromNode() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | utils.cc | 171 const Shape& sharded_shape, Shape* dst_shape) { in DetermineArgumentLayoutsFromCompileOptions()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | winograd_util.cc | 114 OHWI dst_shape; in RearrangeWeightsToWinograd4x4To6x6Weights() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 516 def _broadcast_to_ragged_shape(rt_input, dst_shape, broadcast_inner_dimensions): argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 117 const auto dst_shape = outputs[0]->tensor.shape; in IsGenericAdd() local
723 const auto dst_shape = BHWC(gpu_tensor.Batch(), gpu_tensor.Height(), in GetOutputTensor() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 286 TensorShape dst_shape = in operator ()() local
|
| D | conv_ops_3d.cc | 435 TensorShape dst_shape = in launch() local
|
| D | conv_grad_ops_3d.cc | 1335 TensorShape dst_shape = in Compute() local
1831 TensorShape dst_shape = in Compute() local
|
| D | conv_ops.cc | 912 TensorShape dst_shape = in operator ()() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 97 const auto dst_shape = outputs[0]->tensor.shape; in IsGenericAdd() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 738 ArrayRef<int64_t> dst_shape = in matchAndRewrite() local
|