Searched refs:DeviceTypeString (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | types_test.cc | 27 EXPECT_EQ("CPU", DeviceTypeString(DeviceType(DEVICE_CPU))); in TEST()
28 EXPECT_EQ("GPU", DeviceTypeString(DeviceType(DEVICE_GPU))); in TEST()
|
| D | types.cc | 200 string DeviceTypeString(const DeviceType& device_type) { in DeviceTypeString() function
|
| D | op_kernel.cc | 1250 return strings::StrCat(op_type, ":", DeviceTypeString(device_type), ":", in Key()
1418 const std::string device_str = DeviceTypeString(device_type); in FindKernelDef()
1636 "' OpKernel for '", DeviceTypeString(device_type), in CreateOpKernel()
|
| D | types.h | 108 std::string DeviceTypeString(const DeviceType& device_type);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | inspecting_placer.cc | 61 out->append(DeviceTypeString(type_and_pref.first)); in DebugString()
|
| D | colocation_graph.cc | 79 v.push_back(DeviceTypeString(device_and_type.first)); in DeviceTypeAndPriorityToString()
1358 DeviceTypeString(device_type.first), " "); in DebugInfo()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 67 DeviceTypeString(context->device_type()))); in AvgPoolingOp()
261 DeviceTypeString(context->device_type()))); in AvgPoolingGradOp()
|
| D | pooling_ops_3d.cc | 138 DeviceTypeString(context->device_type()))); in Pooling3DOp()
312 "on device type ", DeviceTypeString(context->device_type()))); in MaxPooling3dGradOp()
466 "on device type ", DeviceTypeString(context->device_type()))); in AvgPooling3dGradOp()
|
| D | maxpooling_op.cc | 232 DeviceTypeString(context->device_type()))); in MaxPoolingGradOp()
465 "on device type ", DeviceTypeString(context->device_type()))); in MaxPoolingGradGradOp()
778 DeviceTypeString(context->device_type()))); in MaxPoolingNoMaskOp()
840 DeviceTypeString(context->device_type()))); in MaxPoolingNoMaskV2Op()
|
| D | pooling_ops_common.h | 107 DeviceTypeString(context->device_type()))); in MaxPoolingOp()
|