Home
last modified time | relevance | path

Searched refs:DataTypeAlwaysOnHost (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dmemory_types.cc71 return (dtype == DT_INT32 || DataTypeAlwaysOnHost(dtype)) ? HOST_MEMORY in MTypeFromDType()
132 if (DataTypeAlwaysOnHost(inp_dtypes[i])) { in MemoryTypesForNode()
137 if (DataTypeAlwaysOnHost(out_dtypes[i])) { in MemoryTypesForNode()
Dtypes.cc214 bool DataTypeAlwaysOnHost(DataType dt) { in DataTypeAlwaysOnHost() function
Dtypes.h483 bool DataTypeAlwaysOnHost(DataType dt);
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc364 if (DataTypeAlwaysOnHost(dtype)) { in ComputeAsync()
371 if (DataTypeAlwaysOnHost(dtype)) { in ComputeAsync()