Searched refs:initialization_values (Results 1 – 2 of 2) sorted by relevance
63 void InitializeTensor(const std::vector<float>& initialization_values, in InitializeTensor() argument67 if (!initialization_values.empty()) { in InitializeTensor()68 for (int i = 0; i < initialization_values.size(); ++i) { in InitializeTensor()69 type_tensor(i) = static_cast<T>(initialization_values[i]); in InitializeTensor()81 InitializeTensor<int32>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()85 InitializeTensor<float>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()89 InitializeTensor<quint8>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()93 InitializeTensor<uint8>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()97 InitializeTensor<bool>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()101 if (!input.initialization_values.empty()) { in CreateTensorsFromInputInfo()[all …]
30 std::vector<float> initialization_values; member