Home
last modified time | relevance | path

Searched refs:MTypeFromDType (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dmemory_types.cc70 MemoryType MTypeFromDType(const DataType dtype) { in MTypeFromDType() function
103 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()
Dtypes.h478 MemoryType MTypeFromDType(const DataType dtype);
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc104 if (MTypeFromDType(type) == HOST_MEMORY) { in UpdateArgAndRetvalMetadata()
115 if (MTypeFromDType(type) == HOST_MEMORY) { in UpdateArgAndRetvalMetadata()
Dfunction.cc580 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()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc143 if (MTypeFromDType(dtype) == HOST_MEMORY) { in FillOutputDevices()
195 } else if (MTypeFromDType(dtype) == HOST_MEMORY) { in Instantiate()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc340 } else if (MTypeFromDType(tensor_handle->dtype) == HOST_MEMORY) { in AddInputDevicesToCacheKey()