Searched refs:output_tensor_size (Results 1 – 1 of 1) sorted by relevance
1610 std::vector<int64> output_tensor_size; in CalculateOutputTensorSize() local1611 output_tensor_size.reserve(op_info.outputs().size()); in CalculateOutputTensorSize()1623 output_tensor_size.push_back(output_size); in CalculateOutputTensorSize()1625 return output_tensor_size; in CalculateOutputTensorSize()