Searched defs:ok_dtypes (Results 1 – 3 of 3) sorted by relevance
99 std::vector<DataType> ok_dtypes; member108 std::vector<DataType> ok_dtypes; member
125 bool IsOkDType(DataType dtype, const std::vector<DataType>* ok_dtypes) { in IsOkDType()159 const std::vector<DataType>* ok_dtypes = nullptr, in ConvertToTensorInPlace()
249 std::vector<DataType>* ok_dtypes = nullptr; in InitializeAttribute() local