| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer_test.cc | 33 constexpr char kDevice[] = "/job:localhost/replica:0/task:0/device:CPU:0"; variable
46 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
47 NDef("y", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
48 NDef("z", "Identity", {"y"}, {{"T", DT_FLOAT}}, kDevice)}, in TEST_F()
71 for (NodeDef& node : *expected.mutable_node()) node.set_device(kDevice); in TEST_F()
115 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
116 NDef("y", "XTimesTwo", {"x"}, {}, kDevice), in TEST_F()
117 NDef("z", "Identity", {"y"}, {{"T", DT_FLOAT}}, kDevice)}, in TEST_F()
165 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
166 NDef("y", "Exp_func", {"x"}, {}, kDevice), in TEST_F()
[all …]
|
| D | meta_optimizer_test.cc | 43 constexpr char kDevice[] = "/device:CPU:0"; variable
294 {NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
295 NDef("b", "Placeholder", {}, {{"dtype", DT_INT32}}, kDevice), in TEST_F()
297 NDef("square", "MySquare", {"a"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
298 NDef("quadratic", "MyQuadratic", {"b"}, {{"T", DT_INT32}}, kDevice), in TEST_F()
300 NDef("out_s", "Identity", {"square:0"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
301 NDef("out_q", "Identity", {"quadratic:0"}, {{"T", DT_INT32}}, kDevice)}, in TEST_F()
443 {NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
444 NDef("b", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
445 NDef("fwd", "Fwd", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
[all …]
|
| D | generic_layout_optimizer_transposer_test.cc | 519 constexpr char kDevice[] = "/device:GPU:0"; in TEST_F() local
522 TF_ASSERT_OK(transposer.CreateConstPermNode(&context, kNodeName, kDevice, in TEST_F()
529 EXPECT_EQ(const_perm_node->GetDevice(), kDevice); in TEST_F()
564 constexpr char kDevice[] = "/device:GPU:0"; in TEST_F() local
570 &context, kNodeNameFormat, DT_DOUBLE, kDevice, input_shape, {0, 3, 1, 2}, in TEST_F()
584 EXPECT_EQ(transpose_node->GetDevice(), kDevice); in TEST_F()
|
| D | constant_folding_test.cc | 4160 constexpr char kDevice[] = "/job:localhost/replica:0/task:0/device:CPU:0"; in TEST_F() local
4184 kDevice), in TEST_F()
4185 NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
4191 kDevice), in TEST_F()
4192 NDef("y", "Identity", {"case"}, {{"T", DT_FLOAT}}, kDevice)}, in TEST_F()
|
| /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/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/framework/ |
| D | function_test.cc | 1385 constexpr char kDevice[] = "/device:CPU:0"; in TEST() local
1389 NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST()
1390 NDef("b", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST()
1391 NDef("x", "Func1", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST()
1396 kDevice), in TEST()
1397 NDef("z", "Func4", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST()
|
| /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 | 34 enum class MemorySpace { kHost, kDevice }; enumerator
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 137 case MemorySpace::kDevice: in MemorySpaceString()
1030 return MemorySpace::kDevice; in GetPointerMemorySpace()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 33 constexpr char kDevice[] = "/device:CPU:0"; variable
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 174 case MemorySpace::kDevice: in MemorySpaceString()
1307 return MemorySpace::kDevice; in GetPointerMemorySpace()
|
| D | cuda_blas.cc | 469 case blas::PointerMode::kDevice: in CUBLASPointerMode()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 133 kDevice, enumerator
|