Lines Matching full:tensor1
905 TENSOR tensor1[CNN_MAX_BRANCHES] = { { 0 } }; in av1_cnn_predict_c() local
919 init_tensor(&tensor1[b]); in av1_cnn_predict_c()
932 assign_tensor(&tensor1[branch], (float **)input, in av1_cnn_predict_c()
935 // Swap tensor1 and tensor2 in av1_cnn_predict_c()
936 swap_tensor(&tensor1[branch], &tensor2[branch]); in av1_cnn_predict_c()
938 i_width = tensor1[branch].width; in av1_cnn_predict_c()
939 i_height = tensor1[branch].height; in av1_cnn_predict_c()
964 if (!copy_active_tensor_to_branches(&tensor1[branch], layer_config, in av1_cnn_predict_c()
970 assert(tensor1[branch].channels == layer_config->in_channels); in av1_cnn_predict_c()
976 convolve_layer_mt((const float **)tensor1[branch].buf, in av1_cnn_predict_c()
977 tensor1[branch].width, tensor1[branch].height, in av1_cnn_predict_c()
978 tensor1[branch].stride, layer_config, thread_data, in av1_cnn_predict_c()
981 av1_cnn_convolve((const float **)tensor1[branch].buf, in av1_cnn_predict_c()
982 tensor1[branch].width, tensor1[branch].height, in av1_cnn_predict_c()
983 tensor1[branch].stride, layer_config, in av1_cnn_predict_c()
987 av1_cnn_deconvolve((const float **)tensor1[branch].buf, in av1_cnn_predict_c()
988 tensor1[branch].width, tensor1[branch].height, in av1_cnn_predict_c()
989 tensor1[branch].stride, layer_config, in av1_cnn_predict_c()
1072 free_tensor(&tensor1[b]); in av1_cnn_predict_c()