Home
last modified time | relevance | path

Searched refs:SetStateVar (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dsoftmax.cc70 src_desc.SetStateVar("BatchedWidth", "true"); in CreateSoftmax()
75 dst_desc.SetStateVar("BatchedWidth", "true"); in CreateSoftmax()
Dpooling.cc33 src_desc.SetStateVar("BatchedWidth", "true"); in GetAveragePoolingKernelCode()
38 dst_desc.SetStateVar("BatchedWidth", "true"); in GetAveragePoolingKernelCode()
150 src_desc.SetStateVar("BatchedWidth", "true"); in GetMaxPoolingKernelCode()
155 dst_desc.SetStateVar("BatchedWidth", "true"); in GetMaxPoolingKernelCode()
161 dst_ind_desc.SetStateVar("BatchedWidth", "true"); in GetMaxPoolingKernelCode()
Dresize.cc44 src_desc.SetStateVar("BatchedWidth", "true"); in GetResizeCode()
49 dst_desc.SetStateVar("BatchedWidth", "true"); in GetResizeCode()
179 src_desc.SetStateVar("BatchedWidth", "true"); in GetResize3DCode()
184 dst_desc.SetStateVar("BatchedWidth", "true"); in GetResize3DCode()
Dmax_unpooling.cc31 src_desc.SetStateVar("BatchedWidth", "true"); in GetMaxUnpoolingKernelCode()
37 src_ind_desc.SetStateVar("BatchedWidth", "true"); in GetMaxUnpoolingKernelCode()
42 dst_desc.SetStateVar("BatchedWidth", "true"); in GetMaxUnpoolingKernelCode()
Dconcat_z.cc130 src_desc.SetStateVar("BatchedWidth", "true"); in CreateConcatZ()
136 dst_desc.SetStateVar("BatchedWidth", "true"); in CreateConcatZ()
Dadd.cc40 src_desc.SetStateVar("BatchedWidth", "true"); in CreateAdd()
Dconv_buffer_1x1.cc172 src_desc.SetStateVar("BatchedWidth", "true"); in GenerateConvBuffer1x1()
175 src_desc.SetStateVar("ElementsX2", "true"); in GenerateConvBuffer1x1()
177 src_desc.SetStateVar("ElementsX4", "true"); in GenerateConvBuffer1x1()
191 dst_desc.SetStateVar("BatchedWidth", "true"); in GenerateConvBuffer1x1()
Dconv_constants.cc105 src_desc.SetStateVar("BatchedWidth", "true"); in GenerateConvolutionConstantCode()
111 dst_desc.SetStateVar("BatchedWidth", "true"); in GenerateConvolutionConstantCode()
Dconvolution_transposed_4x4.cc77 src_desc.SetStateVar("BatchedWidth", "true"); in GenerateConvolutionTransposedCode()
83 dst_desc.SetStateVar("BatchedWidth", "true"); in GenerateConvolutionTransposedCode()
Ddepthwise_conv.cc75 src_desc.SetStateVar("BatchedWidth", "true"); in GenerateDepthwiseConvolutionCode()
84 dst_desc.SetStateVar("BatchedWidth", "true"); in GenerateDepthwiseConvolutionCode()
Dconvolution_transposed_3x3.cc62 src_desc.SetStateVar("BatchedWidth", "true"); in GenerateConvolutionTransposedCode()
68 dst_desc.SetStateVar("BatchedWidth", "true"); in GenerateConvolutionTransposedCode()
Dwinograd.cc279 src_desc.SetStateVar("ACCUM_FLT", cl_type); in GetWinograd4x4To36TileX6Code()
549 src_desc.SetStateVar("ACCUM_FLT", cl_type); in GetWinograd36To4x4Tile4x1Code()
Dconv_metal.cc1002 src_desc.SetStateVar("BatchedWidth", "true"); in CreateConvolutionMetal()
1007 dst_desc.SetStateVar("BatchedWidth", "true"); in CreateConvolutionMetal()
1127 src_desc.SetStateVar("BatchedWidth", "true"); in CreateConvolutionMetalWino4x4To6x6()
1132 dst_desc.SetStateVar("BatchedWidth", "true"); in CreateConvolutionMetalWino4x4To6x6()
Delementwise.cc310 src_desc.SetStateVar("BatchedWidth", "true"); in CreateElementwiseTwoInput()
Dconv_powervr.cc344 src_desc.SetStateVar("BatchedWidth", "true"); in GenerateConv()
406 dst_desc.SetStateVar("BatchedWidth", "true"); in GenerateConv()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc191 src_desc->SetStateVar("BatchedWidth", "true"); in AssembleCode()
199 dst_desc->SetStateVar("BatchedWidth", "true"); in AssembleCode()
Dgpu_object_desc.h112 void SetStateVar(const std::string& key, const std::string& value) const { in SetStateVar() function