Searched refs:td_1 (Results 1 – 1 of 1) sorted by relevance
103 TensorDescriptor td_1; in WinogradFromNode() local104 td_1.storage_type = SelectBestStorageType( in WinogradFromNode()107 td_1.data_type = op_def.src_tensors[0].data_type; in WinogradFromNode()108 td_1.layout = op_def.src_tensors[0].layout; in WinogradFromNode()109 gpu_subgraph->new_tensors = {{shape_0, td_0}, {shape_1, td_1}}; in WinogradFromNode()126 conv_def.dst_tensors.push_back(td_1); in WinogradFromNode()135 winograd_down_def.src_tensors.push_back(td_1); in WinogradFromNode()