Searched refs:transpose_def (Results 1 – 1 of 1) sorted by relevance
250 OperationDef transpose_def; in GPUOperationFromNode() local251 transpose_def.precision = op_def.precision; in GPUOperationFromNode()252 transpose_def.src_tensors.push_back(op_def.src_tensors[1]); in GPUOperationFromNode()253 transpose_def.dst_tensors.push_back(transposed_desc); in GPUOperationFromNode()260 CreateTranspose(transpose_def, transpose_attr)); in GPUOperationFromNode()