Searched refs:initialization_values (Results 1 – 2 of 2) sorted by relevance
64 void InitializeTensor(const std::vector<float>& initialization_values, in InitializeTensor() argument68 if (!initialization_values.empty()) { in InitializeTensor()69 for (int i = 0; i < initialization_values.size(); ++i) { in InitializeTensor()70 type_tensor(i) = static_cast<T>(initialization_values[i]); in InitializeTensor()82 InitializeTensor<int32>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()86 InitializeTensor<int64>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()90 InitializeTensor<float>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()94 InitializeTensor<quint8>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()98 InitializeTensor<uint8>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()102 InitializeTensor<bool>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()[all …]
30 std::vector<float> initialization_values; member