Searched refs:AssignedDevice (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | node_matchers_test.cc | 31 using testing::matchers::AssignedDevice;
154 TEST(NodeMatchers, AssignedDevice) { in TEST() argument
172 NodeWith(AssignedDevice("/job:localhost/replica:0/task:0/device:CPU:0"))); in TEST()
173 EXPECT_THAT(unassigned_add.node(), NodeWith(AssignedDevice(""))); in TEST()
176 NodeWith(AssignedDevice( in TEST()
|
| D | increase_dynamism_for_auto_jit_pass_test.cc | 31 using testing::matchers::AssignedDevice;
123 Op("Sub"), AssignedDevice(kHostName), in TEST()
125 Out(NodeWith(Op("Slice"), AssignedDevice(kHostName), in TEST()
127 Out(NodeWith(Op("Slice"), AssignedDevice(kHostName), in TEST()
130 Op("ConcatV2"), AssignedDevice(kHostName), in TEST()
136 Op("Slice"), AssignedDevice(kDeviceName), in TEST()
|
| D | node_matchers.h | 188 impl::NodeMatcherProperties AssignedDevice(string assigned_device);
|
| D | node_matchers.cc | 445 impl::NodeMatcherProperties AssignedDevice(string assigned_device) { in AssignedDevice() function
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope_internal.h | 63 enum class AssignedDevice; member
81 Impl(const Scope& other, Tags::AssignedDevice, const string& assigned_device);
|
| D | scope.cc | 234 Scope::Impl::Impl(const Scope& other, Tags::AssignedDevice, in Impl() argument
353 builder->AssignedDevice(impl()->assigned_device_); in UpdateBuilder()
454 return Scope(new Impl(*this, Impl::Tags::AssignedDevice(), assigned_device)); in WithAssignedDevice()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | node_builder.h | 106 NodeBuilder& AssignedDevice(StringPiece device);
|
| D | node_builder.cc | 105 NodeBuilder& NodeBuilder::AssignedDevice(StringPiece device) { in AssignedDevice() function in tensorflow::NodeBuilder
|