Searched refs:bias_desc (Results 1 – 7 of 7) sorted by relevance
1049 BufferDescriptor bias_desc; in CreateConvolutionMetal() local1050 bias_desc.element_type = weights_type; in CreateConvolutionMetal()1051 bias_desc.element_size = 4; in CreateConvolutionMetal()1052 bias_desc.memory_type = mem_type; in CreateConvolutionMetal()1053 bias_desc.data = ReorderBiasesForConv( in CreateConvolutionMetal()1056 bias_desc.size = bias_desc.data.size(); in CreateConvolutionMetal()1058 "biases", absl::make_unique<BufferDescriptor>(std::move(bias_desc))); in CreateConvolutionMetal()1164 BufferDescriptor bias_desc; in CreateConvolutionMetalWino4x4To6x6() local1165 bias_desc.element_type = weights_type; in CreateConvolutionMetalWino4x4To6x6()1166 bias_desc.element_size = 4; in CreateConvolutionMetalWino4x4To6x6()[all …]
507 TensorLinearDescriptor bias_desc; in CreateWinograd36To4x4() local508 bias_desc.storage_type = LinearStorageType::BUFFER; in CreateWinograd36To4x4()509 bias_desc.element_type = definition.GetDataType(); in CreateWinograd36To4x4()510 bias_desc.UploadLinearData(biases); in CreateWinograd36To4x4()512 std::move(bias_desc))); in CreateWinograd36To4x4()
49 #define GET_BIAS_DESC_FROM_OP_PD(op_pd) op_pd->bias_desc()103 #define PRIMITIVE_DESC_BIAS bias_desc()
103 BatchDescriptor bias_desc; in RunGpuConvForwardActivation() local104 bias_desc.set_count(1) in RunGpuConvForwardActivation()132 params.config.fusion->side_input_scale, bias_desc, in RunGpuConvForwardActivation()
478 input_bias_ = new memory(mkldnn_matmul_fwd_pd->bias_desc(), in GetBiasHandle()481 new memory(mkldnn_matmul_fwd_pd->bias_desc(), this->cpu_engine_); in GetBiasHandle()
1562 conv_fwd_pd->bias_desc(), &scaled_bias_buf_); in GetBiasHandle()1604 (conv_prim_desc.bias_desc().get_size() / sizeof(Tbias))); in AllocatePersistentTensor()
583 se::dnn::BatchDescriptor bias_desc;584 bias_desc.set_count(1)679 bias_desc, bias_ptr, // bias