Searched refs:out_dtypes (Results 1 – 2 of 2) sorted by relevance
89 DataTypeVector out_dtypes; in MemoryTypesForNode() local91 InOutTypesForNode(ndef, *op_def, &inp_dtypes, &out_dtypes)); in MemoryTypesForNode()104 for (const auto& t : out_dtypes) out_mtypes->push_back(MTypeFromDType(t)); in MemoryTypesForNode()128 CHECK_LE(out_mtypes->size(), out_dtypes.size()); in MemoryTypesForNode()137 if (DataTypeAlwaysOnHost(out_dtypes[i])) { in MemoryTypesForNode()
366 out_dtypes = [dtypes.float32, dtypes.float32]368 out_dtypes = out_dtypes + [dtypes.int32]370 loop_fn, 2, loop_fn_dtypes=out_dtypes)