Searched refs:batch_descs (Results 1 – 1 of 1) sorted by relevance
97 DnnBatchDescriptors batch_descs; in MakeBatchNormDescriptors() local98 batch_descs.input_desc.set_layout(se::dnn::DataLayout::kBatchDepthYX) in MakeBatchNormDescriptors()104 batch_descs.scale_offset_desc.set_layout(se::dnn::DataLayout::kBatchDepthYX) in MakeBatchNormDescriptors()105 .set_feature_map_count(batch_descs.input_desc.feature_map_count()) in MakeBatchNormDescriptors()110 return batch_descs; in MakeBatchNormDescriptors()123 DnnBatchDescriptors batch_descs = in AssignCommonParams() local125 params->operand_desc = batch_descs.input_desc; in AssignCommonParams()126 params->scale_offset_desc = batch_descs.scale_offset_desc; in AssignCommonParams()