| /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
135 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F()
154 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F()
175 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F()
222 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F()
239 TrivialTestGraphInputYielder fake_input(4, 1, 10, false, {kDevice}); in TEST_F()
312 {NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
313 NDef("b", "Placeholder", {}, {{"dtype", DT_INT32}}, kDevice), in TEST_F()
315 NDef("square", "MySquare", {"a"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
316 NDef("quadratic", "MyQuadratic", {"b"}, {{"T", DT_INT32}}, 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/skia/gm/ |
| D | fp_sample_chaining.cpp | 152 kDevice, enumerator
163 case kDevice: in wrap()
231 draw({ kDevice, kUniform }); // Same as identity (uniform applied *before*
233 draw({ kExplicit, kUniform, kDevice }); // Scale Y by 2x and translate up by 16px
234 draw({ kDevice, kExplicit, kUniform, kDevice }); // Identity, again.
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | target_annotation.cc | 96 SetAnnotation(op, kDevice, device, builder); in SetTargetAnnotation()
105 SetAnnotation(op, kDevice, GetHardwareName(hardware), builder); in SetTargetAnnotation()
115 SetAnnotation(op, kDevice, "CPU", builder); in SetTargetAnnotation()
|
| D | get_alternative_subgraph.cc | 236 kDevice, builder->getStringAttr(target_device_inference_type.hardware)); in GetAlternativeViewForSpec()
260 op->setAttr(kDevice, builder->getStringAttr( in GetAlternativeViewForSpec()
290 auto device_attr = func->getAttrOfType<StringAttr>(kDevice); in runOnOperation()
|
| D | raise_target_subgraphs.cc | 221 kDevice, in BuildFuncOp()
308 call_op->setAttr(kDevice, in ExtractSubgraphToFunc()
|
| D | pick_subgraphs.cc | 443 kDevice, in RewireSubgraphs()
|
| /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/compiler/mlir/tfrt/transforms/ |
| D | remove_device_attribute.cc | 31 constexpr const char* kDevice = "device"; variable
57 if (attr.first != kDevice) { in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/common/ |
| D | targets.h | 35 constexpr char kDevice[] = "tac.device"; variable
90 auto device = op->getAttrOfType<StringAttr>(kDevice); in GetTargetAnnotation()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_cudamalloc_allocator.h | 40 return AllocatorMemoryType::kDevice; in GetMemoryType()
|
| D | gpu_cudamallocasync_allocator.h | 97 return AllocatorMemoryType::kDevice; in GetMemoryType()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_mem_allocator.h | 74 return AllocatorMemoryType::kDevice; in GetMemoryType()
|
| /external/perfetto/src/traced/probes/ps/ |
| D | process_stats_data_source_unittest.cc | 339 const char kDevice[] = "/dev/dummy"; in TEST_F() local
356 EXPECT_EQ(symlink(kDevice, link.c_str()), 0) in TEST_F()
357 << "symlink('" << kDevice << "','" << link.c_str() << "') failed"; in TEST_F()
424 ASSERT_EQ(fd_path.path(), kDevice); in TEST_F()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 1452 constexpr char kDevice[] = "/device:CPU:0"; in TEST() local
1456 NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST()
1457 NDef("b", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST()
1458 NDef("x", "Func1", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST()
1463 kDevice), in TEST()
1464 NDef("z", "Func4", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST()
|
| D | allocator.h | 104 kDevice = 1, // Memory on device. enumerator
|
| /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/cronet/net/base/ |
| D | network_interfaces_fuchsia.cc | 124 case fuchsia::net::interfaces::DeviceClass::kDevice: in ConvertConnectionType()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_blas_lt.h | 85 kDevice, enumerator
|
| D | cuda_blas_lt.cc | 88 case BlasLt::PointerMode::kDevice: in AsCublasLtPointerMode()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/ |
| D | execution_metadata_exporter.cc | 68 auto device = op->getAttrOfType<mlir::StringAttr>(mlir::TFL::tac::kDevice); in GetDeviceName()
|
| /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/compiler/xla/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()
1040 return MemorySpace::kDevice; in GetPointerMemorySpace()
|