Searched refs:MTypeFromDType (Results 1 – 6 of 6) sorted by relevance
70 MemoryType MTypeFromDType(const DataType dtype) { in MTypeFromDType() function103 for (const auto& t : inp_dtypes) inp_mtypes->push_back(MTypeFromDType(t)); in MemoryTypesForNode()104 for (const auto& t : out_dtypes) out_mtypes->push_back(MTypeFromDType(t)); in MemoryTypesForNode()
478 MemoryType MTypeFromDType(const DataType dtype);
104 if (MTypeFromDType(type) == HOST_MEMORY) { in UpdateArgAndRetvalMetadata()115 if (MTypeFromDType(type) == HOST_MEMORY) { in UpdateArgAndRetvalMetadata()
580 input_memory_types.push_back(MTypeFromDType(t)); in CreateKernel()584 output_memory_types.push_back(MTypeFromDType(t)); in CreateKernel()982 if (MTypeFromDType(arg_type) == HOST_MEMORY) { in RunRemote()989 if (MTypeFromDType(ret_type) == HOST_MEMORY) { in RunRemote()
143 if (MTypeFromDType(dtype) == HOST_MEMORY) { in FillOutputDevices()195 } else if (MTypeFromDType(dtype) == HOST_MEMORY) { in Instantiate()
340 } else if (MTypeFromDType(tensor_handle->dtype) == HOST_MEMORY) { in AddInputDevicesToCacheKey()