Searched refs:DEVICE_DEFAULT (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack_ops.cc | 45 Name("Stack").Device(DEVICE_DEFAULT).HostMemory("handle"), StackOp);
49 .Device(DEVICE_DEFAULT)
64 .Device(DEVICE_DEFAULT) \
69 .Device(DEVICE_DEFAULT) \
74 .Device(DEVICE_DEFAULT) \
79 .Device(DEVICE_DEFAULT) \
92 .Device(DEVICE_DEFAULT) \
99 .Device(DEVICE_DEFAULT) \
106 .Device(DEVICE_DEFAULT) \
112 .Device(DEVICE_DEFAULT) \
[all …]
|
| D | queue_ops.cc | 60 Name("QueueEnqueueV2").Device(DEVICE_DEFAULT).HostMemory("handle"),
63 Name("QueueDequeueV2").Device(DEVICE_DEFAULT).HostMemory("handle"),
66 Name("QueueCloseV2").Device(DEVICE_DEFAULT).HostMemory("handle"),
69 .Device(DEVICE_DEFAULT)
74 Name("QueueIsClosedV2").Device(DEVICE_DEFAULT).HostMemory("handle"),
|
| D | identity_op.cc | 51 Name("Identity").Device(DEVICE_DEFAULT).TypeConstraint("T", DT_STRING),
54 Name("Identity").Device(DEVICE_DEFAULT).TypeConstraint<Variant>("T"),
57 .Device(DEVICE_DEFAULT)
|
| D | sendrecv_ops.cc | 126 REGISTER_KERNEL_BUILDER(Name("_Send").Device(DEVICE_DEFAULT), SendOp);
132 REGISTER_KERNEL_BUILDER(Name("Send").Device(DEVICE_DEFAULT), SendOp);
135 Name("_HostSend").Device(DEVICE_DEFAULT).HostMemory("tensor"), SendOp);
220 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_DEFAULT), RecvOp);
226 REGISTER_KERNEL_BUILDER(Name("Recv").Device(DEVICE_DEFAULT), RecvOp);
229 Name("_HostRecv").Device(DEVICE_DEFAULT).HostMemory("tensor"), RecvOp);
|
| D | control_flow_ops.cc | 56 Name("Switch").Device(DEVICE_DEFAULT).HostMemory("pred"), SwitchOp);
61 Name("_SwitchN").Device(DEVICE_DEFAULT).HostMemory("output_index"),
238 Name("Merge").Device(DEVICE_DEFAULT).HostMemory("value_index"), MergeOp);
303 REGISTER_KERNEL_BUILDER(Name("Enter").Device(DEVICE_DEFAULT), EnterOp);
360 REGISTER_KERNEL_BUILDER(Name("Exit").Device(DEVICE_DEFAULT), ExitOp);
412 REGISTER_KERNEL_BUILDER(Name("NextIteration").Device(DEVICE_DEFAULT),
475 .Device(DEVICE_DEFAULT)
486 REGISTER_KERNEL_BUILDER(Name("ControlTrigger").Device(DEVICE_DEFAULT),
|
| D | no_op.cc | 20 REGISTER_KERNEL_BUILDER(Name("NoOp").Device(DEVICE_DEFAULT), NoOp);
|
| D | identity_n_op.cc | 27 REGISTER_KERNEL_BUILDER(Name("IdentityN").Device(DEVICE_DEFAULT), IdentityNOp);
|
| D | fifo_queue_op.cc | 24 Name("FIFOQueueV2").Device(DEVICE_DEFAULT).HostMemory("handle"),
|
| D | host_constant_op.cc | 61 Name("HostConst").Device(DEVICE_DEFAULT).HostMemory("output"),
|
| D | constant_op.cc | 358 REGISTER_KERNEL_BUILDER(Name("Placeholder").Device(DEVICE_DEFAULT),
360 REGISTER_KERNEL_BUILDER(Name("PlaceholderV2").Device(DEVICE_DEFAULT),
|
| D | logging_ops.cc | 76 .Device(DEVICE_DEFAULT)
|
| D | collective_ops.cc | 469 : AsyncOpKernel(c), device_type_(DEVICE_DEFAULT) { in CollectiveReduceV2OpKernel()
619 : AsyncOpKernel(c), device_type_(DEVICE_DEFAULT) { in CollectiveGatherV2OpKernel()
754 : AsyncOpKernel(c), device_type_(DEVICE_DEFAULT) { in CollectiveBcastSendV2OpKernel()
880 : AsyncOpKernel(c), device_type_(DEVICE_DEFAULT) { in CollectiveBcastRecvV2OpKernel()
|
| D | cast_op.cc | 284 Name("_HostCast").Device(DEVICE_DEFAULT).HostMemory("x").HostMemory("y"),
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | types.cc | 38 const char* const DEVICE_DEFAULT = "DEFAULT"; variable
|
| D | types.h | 74 TF_EXPORT extern const char* const DEVICE_DEFAULT; // "DEFAULT"
|
| D | op_kernel.cc | 1349 const string default_key = Key(node_op, DEVICE_DEFAULT, label); in FindKernelRegistration()
|