Searched refs:MTypeFromDType (Results 1 – 6 of 6) sorted by relevance
133 : MTypeFromDType(type); in UpdateArgAndRetvalMetadata()150 : MTypeFromDType(type); in UpdateArgAndRetvalMetadata()
650 input_memory_types.push_back(MTypeFromDType(t)); in CreateKernel()654 output_memory_types.push_back(MTypeFromDType(t)); in CreateKernel()1030 if (MTypeFromDType(arg_type) == HOST_MEMORY) { in RunRemote()1037 if (MTypeFromDType(ret_type) == HOST_MEMORY) { in RunRemote()
73 MemoryType MTypeFromDType(const DataType dtype) { in MTypeFromDType() function
525 MemoryType MTypeFromDType(const DataType dtype);
665 } else if (MTypeFromDType(dtype) == HOST_MEMORY) { in Instantiate()
295 : MTypeFromDType(tensor_handle->dtype); in GetDeviceForInput()