Home
last modified time | relevance | path

Searched refs:weights_desc (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_transposed_selector.cc98 const OperationDef& op_def, WeightsDescription* weights_desc) { in SelectConvolutionTransposedWithDynamicWeights() argument
104 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
109 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
117 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
123 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
128 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
133 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
139 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionTransposedWithDynamicWeights()
Dconvolution_selector.cc62 WeightsDescription* weights_desc) { in SelectConvolutionDynamicWeightsAdreno() argument
65 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionDynamicWeightsAdreno()
120 WeightsDescription* weights_desc) { in SelectConvolutionDynamicWeightsMali() argument
125 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionDynamicWeightsMali()
130 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionDynamicWeightsMali()
187 WeightsDescription* weights_desc) { in SelectConvolutionWithDynamicWeights() argument
198 weights_desc); in SelectConvolutionWithDynamicWeights()
202 weights_desc); in SelectConvolutionWithDynamicWeights()
206 *weights_desc = conv.GetWeightsDescription(); in SelectConvolutionWithDynamicWeights()
212 const WeightsDescription& weights_desc, const OperationDef& op_def, in SelectConverterToConvWeights() argument
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter.h30 const WeightsDescription& weights_desc);
42 const OperationDef& op_def, const WeightsDescription& weights_desc);
52 const OperationDef& definition, const WeightsDescription& weights_desc);
Dconv_metal.cc594 const WeightsDescription& weights_desc, const DataType& weights_type) { in ReorderWeightsForConv() argument
596 GetTotalElementsCountForLayout(weights_desc, weights.shape); in ReorderWeightsForConv()
598 RearrangeWeights(weights, weights_desc, weights_type, absl::MakeSpan(result)); in ReorderWeightsForConv()
1032 BufferDescriptor weights_desc; in CreateConvolutionMetal() local
1033 weights_desc.element_type = definition.src_tensors[1].data_type; in CreateConvolutionMetal()
1034 weights_desc.element_size = 4; in CreateConvolutionMetal()
1035 weights_desc.memory_type = mem_type; in CreateConvolutionMetal()
1036 desc.AddSrcBuffer("weights", weights_desc); in CreateConvolutionMetal()
1038 BufferDescriptor weights_desc; in CreateConvolutionMetal() local
1039 weights_desc.element_type = weights_type; in CreateConvolutionMetal()
[all …]
Dconv_weights_converter.cc28 const OperationDef& definition, const WeightsDescription& weights_desc) in ConverterToConvWeights() argument
29 : GPUOperation(definition), weights_desc_(weights_desc) { in ConverterToConvWeights()
188 const OperationDef& definition, const WeightsDescription& weights_desc) { in CreateConverterToConvWeights() argument
189 return ConverterToConvWeights(definition, weights_desc); in CreateConverterToConvWeights()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc216 TensorDescriptor weights_desc = {op_def.src_tensors[1].data_type, in GPUOperationFromNode() local
226 conv_def.src_tensors[1] = weights_desc; in GPUOperationFromNode()
238 weights_desc}, in GPUOperationFromNode()
243 converter_def.dst_tensors.push_back(weights_desc); in GPUOperationFromNode()
293 TensorDescriptor weights_desc = {op_def.src_tensors[1].data_type, in GPUOperationFromNode() local
303 conv_def.src_tensors[1] = weights_desc; in GPUOperationFromNode()
316 weights_desc}}; in GPUOperationFromNode()
320 converter_def.dst_tensors.push_back(weights_desc); in GPUOperationFromNode()
348 WeightsDescription weights_desc; in GPUOperationFromNode() local
350 attr, gpu_info, op_def, &weights_desc); in GPUOperationFromNode()
[all …]
Dconvolution_selector.h42 ModelHints hints, WeightsDescription* weights_desc);
45 const WeightsDescription& weights_desc, const OperationDef& op_def,
Dconvolution_transposed_selector.h35 const OperationDef& op_def, WeightsDescription* weights_desc);
/external/tensorflow/tensorflow/core/util/
Dmkl_types.h55 #define GET_WEIGHTS_DESC_FROM_OP_PD(op_pd) op_pd->weights_desc()
63 filter_md != op_pd->weights_desc()
66 weights_md != op_pd->weights_desc()
110 #define PRIMITIVE_DESC_WEIGHTS weights_desc()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_op_fused.cc226 if (weight_md != matmul_pd->weights_desc()) { in Compute()
235 this->GetCachedWeight(ctx, matmul_pd->weights_desc()); in Compute()
245 weight_mkl.CheckReorderToOpMem(matmul_pd.get()->weights_desc(), in Compute()
Dmkl_conv_ops.cc288 context_.filter_mem.reset(new memory(context_.fwd_pd.get()->weights_desc(), in Setup()
691 if (filter_md != conv_fwd_pd->weights_desc()) { in Compute()
703 filter_data = GetCachedFilter(context, conv_fwd_pd->weights_desc()); in Compute()
709 filter.CheckReorderToOpMem(conv_fwd_pd->weights_desc(), cpu_engine_, in Compute()
713 conv_fwd_pd->weights_desc(), in Compute()
981 (conv_prim_desc.weights_desc().get_size() / sizeof(Tfilter))); in AllocatePersistentTensor()
991 memory::desc weights_desc = conv_prim_desc.weights_desc(); in AllocatePersistentTensor() local
995 cached_filter_md_shape.AddDim(sizeof(weights_desc) / sizeof(uint8)); in AllocatePersistentTensor()
1000 weights_desc; in AllocatePersistentTensor()
1014 auto filter_md = conv_prim_desc.weights_desc(); in AllocateFilterOutputTensor()
[all …]
Dmkl_conv_grad_input_ops.cc222 context_.bwd_input_pd.get()->weights_desc(), cpu_engine_, DummyData)); in Setup()
462 if (fwd_filter_md != bwd_input_pd->weights_desc()) { in Compute()
464 filter.CheckReorderToOpMem(bwd_input_pd.get()->weights_desc(), in Compute()
Dmkl_qmatmul_op.cc260 if (weight_md != matmul_fwd_pd->weights_desc()) { in Compute()
274 this->GetCachedWeight(context, matmul_fwd_pd->weights_desc()); in Compute()
280 weight.CheckReorderToOpMem(matmul_fwd_pd.get()->weights_desc(), in Compute()
Dmkl_matmul_ops_common.h267 context_.weight_mem.reset(new memory(context_.fwd_pd.get()->weights_desc(), in Setup()
440 weight.CheckReorderToOpMem(matmul_fwd_pd.get()->weights_desc(), cpu_engine_, in CacheWeight()
446 size_t weight_size = matmul_fwd_pd.get()->weights_desc().get_size(); in CacheWeight()
458 auto expected_md = matmul_fwd_pd->weights_desc(); in CacheWeight()
Dmkl_fused_batch_norm_op.cc532 auto weights_desc = memory::desc({2, bwdParams.depth}, MklDnnType<U>(), in Setup() local
534 auto diff_weights_desc = weights_desc; in Setup()
563 new memory(weights_desc, cpu_engine_, DummyData)); in Setup()