Searched refs:CreateConvolutionMetal (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_metal.h | 82 friend ConvolutionMetal CreateConvolutionMetal(
95 ConvolutionMetal CreateConvolutionMetal(const OperationDef& definition,
|
| D | conv_metal.cc | 986 ConvolutionMetal CreateConvolutionMetal(const OperationDef& definition, in CreateConvolutionMetal() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | conv_test.mm | 68 CreateConvolutionMetal(op_def, BHWC(1, 2, 2, 2), attr, env->GetGpuInfo());
104 CreateConvolutionMetal(op_def, BHWC(1, 1, 1, 1), attr, env->GetGpuInfo());
140 CreateConvolutionMetal(op_def, BHWC(1, 1, 1, 1), attr, env->GetGpuInfo());
176 CreateConvolutionMetal(op_def, BHWC(1, 2, 1, 2), attr, env->GetGpuInfo());
212 CreateConvolutionMetal(op_def, BHWC(1, 2, 2, 1), attr, env->GetGpuInfo());
270 auto gpu_op0 = CreateConvolutionMetal(op_def, dst_shape, attr, env->GetGpuInfo());
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | convolution_selector.cc | 143 CreateConvolutionMetal(op_def, dst_shape, attr, gpu_info); in SelectConvolution()
193 CreateConvolutionMetal(op_def, dst_shape, attr_copy, gpu_info); in SelectConvolutionWithDynamicWeights()
|
| D | fully_connected_selector.cc | 98 CreateConvolutionMetal(op_def, dst_shape, conv_attr, gpu_info); in SelectFullyConnected()
|