Searched refs:ints_on_device (Results 1 – 1 of 1) sorted by relevance
156 bool ints_on_device) { in BuildInputArgIndex() argument175 if (ints_on_device && dtypes[i] == DataType::DT_INT32) { in BuildInputArgIndex()314 bool ints_on_device, int* ret_index) { in AddReturnNode() argument340 if (ints_on_device && dtypes[i] == DataType::DT_INT32) { in AddReturnNode()690 bool ints_on_device = in InstantiateFunction() local697 s = helper.BuildInputArgIndex(arg_def, attr_values, ints_on_device); in InstantiateFunction()752 s = helper.AddReturnNode(ret_def, attr_values, fdef.ret(), ints_on_device, in InstantiateFunction()