Home
last modified time | relevance | path

Searched defs:weights_desc (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc62 WeightsDescription* weights_desc) { in SelectConvolutionDynamicWeightsAdreno()
120 WeightsDescription* weights_desc) { in SelectConvolutionDynamicWeightsMali()
187 WeightsDescription* weights_desc) { in SelectConvolutionWithDynamicWeights()
212 const WeightsDescription& weights_desc, const OperationDef& op_def, in SelectConverterToConvWeights()
Dconvolution_transposed_selector.cc98 const OperationDef& op_def, WeightsDescription* weights_desc) { in SelectConvolutionTransposedWithDynamicWeights()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter.cc28 const OperationDef& definition, const WeightsDescription& weights_desc) in ConverterToConvWeights()
188 const OperationDef& definition, const WeightsDescription& weights_desc) { in CreateConverterToConvWeights()
Dconv_metal.cc594 const WeightsDescription& weights_desc, const DataType& weights_type) { in ReorderWeightsForConv()
1032 BufferDescriptor weights_desc; in CreateConvolutionMetal() local
1038 BufferDescriptor weights_desc; in CreateConvolutionMetal() local
1155 BufferDescriptor weights_desc; in CreateConvolutionMetalWino4x4To6x6() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc216 TensorDescriptor weights_desc = {op_def.src_tensors[1].data_type, in GPUOperationFromNode() local
293 TensorDescriptor weights_desc = {op_def.src_tensors[1].data_type, in GPUOperationFromNode() local
348 WeightsDescription weights_desc; in GPUOperationFromNode() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc532 auto weights_desc = memory::desc({2, bwdParams.depth}, MklDnnType<U>(), in Setup() local
Dmkl_conv_ops.cc991 memory::desc weights_desc = conv_prim_desc.weights_desc(); in AllocatePersistentTensor() local