Searched refs:kernel_tensor (Results 1 – 1 of 1) sorted by relevance
503 TF_Tensor *kernel_tensor = NULL, *biases_tensor = NULL; in add_conv_layer() local520 kernel_tensor = TF_AllocateTensor(TF_FLOAT, dims, dims_len, size * sizeof(float)); in add_conv_layer()521 memcpy(TF_TensorData(kernel_tensor), params->kernel, size * sizeof(float)); in add_conv_layer()522 TF_SetAttrTensor(op_desc, "value", kernel_tensor, tf_model->status); in add_conv_layer()611 TF_DeleteTensor(kernel_tensor); in add_conv_layer()