| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer_test.cc | 33 constexpr char kDevice[] = "/device:CPU:0"; variable
51 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
52 NDef("y", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
53 NDef("z", "Identity", {"y"}, {{"T", DT_FLOAT}}, kDevice)}, in TEST_F()
67 EXPECT_EQ(kDevice, node.device()); in TEST_F()
73 EXPECT_EQ(kDevice, node.device()); in TEST_F()
79 EXPECT_EQ(kDevice, node.device()); in TEST_F()
85 EXPECT_EQ(kDevice, node.device()); in TEST_F()
89 EXPECT_EQ(kDevice, node.device()); in TEST_F()
96 EXPECT_EQ(kDevice, node.device()); in TEST_F()
[all …]
|
| D | meta_optimizer_test.cc | 40 constexpr char kDevice[] = "/device:CPU:0"; variable
289 {NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
290 NDef("b", "Placeholder", {}, {{"dtype", DT_INT32}}, kDevice), in TEST_F()
292 NDef("square", "MySquare", {"a"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
293 NDef("quadratic", "MyQuadratic", {"b"}, {{"T", DT_INT32}}, kDevice), in TEST_F()
295 NDef("out_s", "Identity", {"square:0"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
296 NDef("out_q", "Identity", {"quadratic:0"}, {{"T", DT_INT32}}, kDevice)}, in TEST_F()
445 {NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
446 NDef("b", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
447 NDef("fwd", "Fwd", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
[all …]
|
| /external/dynamic_depth/internal/dynamic_depth/ |
| D | const.cc | 13 constexpr char kDevice[] = "Device"; variable
46 const char* DynamicDepthConst::Device() { return kDevice; } in Device()
82 LOG(WARNING) << kPose << " maps to " << kDevice << ", " << kCamera in Namespace()
101 return kDevice; in Namespace()
107 return kDevice; in Namespace()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view_test.cc | 213 constexpr char kDevice[] = "/device:foo:0"; in TEST() local
226 graph.UpdateNode("bar_2", "IdentityN", kDevice, {{"T", list_value}})); in TEST()
229 CheckNode(graph, "bar_2", "IdentityN", kDevice, {{"T", list_value}}, in TEST()
241 constexpr char kDevice[] = "/device:foo:0"; in TEST() local
251 TF_EXPECT_OK(graph.UpdateNode("bar_2", "NotImportant", kDevice, {})); in TEST()
254 CheckNode(graph, "bar_2", "NotImportant", kDevice, {}, {"bar_1:1"}, in TEST()
265 constexpr char kDevice[] = "/device:foo:0"; in TEST() local
273 TF_EXPECT_OK(graph.UpdateNode("foo", "Switch", kDevice, {})); in TEST()
275 CheckNode(graph, "foo", "Switch", kDevice, {}, {}, {"bar"}); in TEST()
282 constexpr char kDevice[] = "/device:foo:0"; in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 1379 constexpr char kDevice[] = "/device:CPU:0"; in TEST() local
1383 NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST()
1384 NDef("b", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST()
1385 NDef("x", "Func1", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST()
1390 kDevice), in TEST()
1391 NDef("z", "Func4", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST()
|
| /external/skqp/src/core/ |
| D | SkRemoteGlyphCache.cpp | 59 enum DescriptorType : bool { kKey = false, kDevice = true }; enumerator
65 bool enableTypefaceFiltering = (type == kDevice); in create_descriptor()
346 kDevice, paint, font, matrix, props, flags, &deviceAutoDesc, &deviceEffects); in getOrCreateCache()
372 kDevice, paint, font, matrix, props, flags, &deviceAutoDesc, &deviceEffects); in getOrCreateCache()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_utils.cc | 344 static const char* const kDevice = "device: which device is placed on."; variable
404 helps.push_back(kDevice); in QueryDoc()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | device_name_utils_test.cc | 37 enum NamePart { kJob = 0x01, kReplica = 0x02, kTask = 0x04, kDevice = 0x08 }; enumerator
54 if (parts_to_test & kDevice) { in RoundTripPartialName()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 37 enum class MemorySpace { kHost, kDevice }; enumerator
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 136 case MemorySpace::kDevice: in MemorySpaceString()
1093 return MemorySpace::kDevice; in GetPointerMemorySpace()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 160 case MemorySpace::kDevice: in MemorySpaceString()
1302 return MemorySpace::kDevice; in GetPointerMemorySpace()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 33 constexpr char kDevice[] = "/device:CPU:0"; variable
|