Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc97 TensorDescriptor td_0; in WinogradFromNode() local
98 td_0.storage_type = SelectBestStorageType( in WinogradFromNode()
101 td_0.data_type = op_def.src_tensors[0].data_type; in WinogradFromNode()
102 td_0.layout = op_def.src_tensors[0].layout; in WinogradFromNode()
109 gpu_subgraph->new_tensors = {{shape_0, td_0}, {shape_1, td_1}}; in WinogradFromNode()
116 winograd_up_def.dst_tensors.push_back(td_0); in WinogradFromNode()
125 conv_def.src_tensors.push_back(td_0); in WinogradFromNode()