Searched refs:out_dtypes (Results 1 – 2 of 2) sorted by relevance
96 DataTypeVector out_dtypes; in MemoryTypesForNode() local98 InOutTypesForNode(ndef, *op_def, &inp_dtypes, &out_dtypes)); in MemoryTypesForNode()140 out_mtypes->resize(out_dtypes.size(), DEVICE_MEMORY); in MemoryTypesForNode()143 CHECK_LE(out_mtypes->size(), out_dtypes.size()); in MemoryTypesForNode()152 if (host_memory_required(out_dtypes[i])) { in MemoryTypesForNode()
483 out_dtypes = [dtypes.float32, dtypes.float32]485 out_dtypes = out_dtypes + [dtypes.int32]