| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| periodic_resample_op.cc | 22 REGISTER_KERNEL_BUILDER(Name("PeriodicResample").Device(DEVICE_CPU),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_logical_and.cc | 19 REGISTER_KERNEL_BUILDER(Name("LogicalAnd").Device(DEVICE_CPU),
|
| cwise_op_logical_not.cc | 19 REGISTER_KERNEL_BUILDER(Name("LogicalNot").Device(DEVICE_CPU),
|
| cwise_op_logical_or.cc | 19 REGISTER_KERNEL_BUILDER(Name("LogicalOr").Device(DEVICE_CPU),
|
| identity_n_op.cc | 26 REGISTER_KERNEL_BUILDER(Name("IdentityN").Device(DEVICE_CPU), IdentityNOp);
|
| no_op.cc | 20 REGISTER_KERNEL_BUILDER(Name("NoOp").Device(DEVICE_CPU), NoOp);
|
| identity_op.cc | 26 REGISTER_KERNEL_BUILDER(Name("Identity").Device(DEVICE_CPU), IdentityOp);
29 REGISTER_KERNEL_BUILDER(Name("StopGradient").Device(DEVICE_CPU), IdentityOp);
32 REGISTER_KERNEL_BUILDER(Name("PreventGradient").Device(DEVICE_CPU), IdentityOp);
36 REGISTER_KERNEL_BUILDER(Name("PlaceholderWithDefault").Device(DEVICE_CPU),
39 REGISTER_KERNEL_BUILDER(Name("RefIdentity").Device(DEVICE_CPU), IdentityOp);
43 REGISTER_KERNEL_BUILDER(Name("DebugGradientIdentity").Device(DEVICE_CPU),
45 REGISTER_KERNEL_BUILDER(Name("DebugGradientRefIdentity").Device(DEVICE_CPU),
|
| fifo_queue_op.cc | 60 REGISTER_KERNEL_BUILDER(Name("FIFOQueue").Device(DEVICE_CPU), FIFOQueueOp);
61 REGISTER_KERNEL_BUILDER(Name("FIFOQueueV2").Device(DEVICE_CPU), FIFOQueueOp);
|
| reduction_ops_all.cc | 23 .Device(DEVICE_CPU)
29 .Device(DEVICE_CPU)
|
| reduction_ops_any.cc | 23 .Device(DEVICE_CPU)
29 .Device(DEVICE_CPU)
|
| debug_ops.cc | 27 REGISTER_KERNEL_BUILDER(Name("Copy").Device(DEVICE_CPU), CopyOp);
29 REGISTER_KERNEL_BUILDER(Name("CopyHost").Device(DEVICE_CPU), CopyOp);
52 REGISTER_KERNEL_BUILDER(Name("DebugIdentity").Device(DEVICE_CPU),
74 Name("DebugNanCount").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
106 .Device(DEVICE_CPU) \
|
| reader_ops.cc | 99 REGISTER_KERNEL_BUILDER(Name("ReaderRead").Device(DEVICE_CPU), ReaderReadOp);
100 REGISTER_KERNEL_BUILDER(Name("ReaderReadV2").Device(DEVICE_CPU), ReaderReadOp);
151 REGISTER_KERNEL_BUILDER(Name("ReaderReadUpTo").Device(DEVICE_CPU),
153 REGISTER_KERNEL_BUILDER(Name("ReaderReadUpToV2").Device(DEVICE_CPU),
169 REGISTER_KERNEL_BUILDER(Name("ReaderNumRecordsProduced").Device(DEVICE_CPU),
171 REGISTER_KERNEL_BUILDER(Name("ReaderNumRecordsProducedV2").Device(DEVICE_CPU),
187 REGISTER_KERNEL_BUILDER(Name("ReaderNumWorkUnitsCompleted").Device(DEVICE_CPU),
190 Name("ReaderNumWorkUnitsCompletedV2").Device(DEVICE_CPU),
207 REGISTER_KERNEL_BUILDER(Name("ReaderSerializeState").Device(DEVICE_CPU),
209 REGISTER_KERNEL_BUILDER(Name("ReaderSerializeStateV2").Device(DEVICE_CPU),
[all...] |
| padding_fifo_queue_op.cc | 68 REGISTER_KERNEL_BUILDER(Name("PaddingFIFOQueue").Device(DEVICE_CPU),
70 REGISTER_KERNEL_BUILDER(Name("PaddingFIFOQueueV2").Device(DEVICE_CPU),
|
| priority_queue_op.cc | 65 REGISTER_KERNEL_BUILDER(Name("PriorityQueue").Device(DEVICE_CPU),
67 REGISTER_KERNEL_BUILDER(Name("PriorityQueueV2").Device(DEVICE_CPU),
|
| snapshot_op.cc | 28 Name("Snapshot").Device(DEVICE_CPU).TypeConstraint<TYPE>("T"), \
|
| sparse_reshape_op.cc | 44 REGISTER_KERNEL_BUILDER(Name("SparseReshape").Device(DEVICE_CPU),
|
| string_to_hash_bucket_op.cc | 62 REGISTER_KERNEL_BUILDER(Name("StringToHashBucket").Device(DEVICE_CPU),
65 REGISTER_KERNEL_BUILDER(Name("StringToHashBucketFast").Device(DEVICE_CPU),
68 REGISTER_KERNEL_BUILDER(Name("StringToHashBucketStrong").Device(DEVICE_CPU),
|
| queue_ops.cc | 113 REGISTER_KERNEL_BUILDER(Name("QueueEnqueue").Device(DEVICE_CPU), EnqueueOp);
114 REGISTER_KERNEL_BUILDER(Name("QueueEnqueueV2").Device(DEVICE_CPU), EnqueueOp);
167 REGISTER_KERNEL_BUILDER(Name("QueueEnqueueMany").Device(DEVICE_CPU),
169 REGISTER_KERNEL_BUILDER(Name("QueueEnqueueManyV2").Device(DEVICE_CPU),
218 REGISTER_KERNEL_BUILDER(Name("QueueDequeue").Device(DEVICE_CPU), DequeueOp);
219 REGISTER_KERNEL_BUILDER(Name("QueueDequeueV2").Device(DEVICE_CPU), DequeueOp);
284 REGISTER_KERNEL_BUILDER(Name("QueueDequeueMany").Device(DEVICE_CPU),
286 REGISTER_KERNEL_BUILDER(Name("QueueDequeueManyV2").Device(DEVICE_CPU),
370 REGISTER_KERNEL_BUILDER(Name("QueueDequeueUpTo").Device(DEVICE_CPU),
372 REGISTER_KERNEL_BUILDER(Name("QueueDequeueUpToV2").Device(DEVICE_CPU),
[all...] |
| guarantee_const_op.cc | 43 REGISTER_KERNEL_BUILDER(Name("GuaranteeConst").Device(DEVICE_CPU),
|
| /external/tensorflow/tensorflow/examples/adding_an_op/ |
| zero_out_op_kernel_2.cc | 86 .Device(DEVICE_CPU)
90 .Device(DEVICE_CPU)
94 .Device(DEVICE_CPU)
100 Name("ZeroOut2").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
111 Name("ZeroOut3").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
|
| /external/tensorflow/tensorflow/python/client/ |
| test_construction_fails_op.cc | 34 REGISTER_KERNEL_BUILDER(Name("ConstructionFails").Device(DEVICE_CPU),
|
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel_test.cc | 48 REGISTER_KERNEL_BUILDER(Name("Test1").Device(tensorflow::DEVICE_CPU),
87 Name("Test3").Device(DEVICE_CPU).TypeConstraint<int8>("T"), TestOp3Cpu);
102 REGISTER_KERNEL_BUILDER(Name("Test4").Device(DEVICE_CPU), DummyKernel);
106 return {DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)};
171 ExpectSuccess("Test1", DEVICE_CPU, {DT_FLOAT, DT_INT32}, {DT_UINT8});
172 ExpectSuccess("Test1", DEVICE_CPU, {DT_FLOAT_REF, DT_INT32}, {DT_UINT8});
182 ExpectSuccess("Test3", DEVICE_CPU, {DT_INT8, DT_INT8}, {});
191 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0]);
202 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0]);
228 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[1])
[all...] |
| /external/tensorflow/tensorflow/core/user_ops/ |
| fact.cc | 40 REGISTER_KERNEL_BUILDER(Name("Fact").Device(tensorflow::DEVICE_CPU), FactOp);
|
| /external/tensorflow/tensorflow/user_ops/ |
| ackermann_op.cc | 44 REGISTER_KERNEL_BUILDER(Name("Ackermann").Device(DEVICE_CPU), AckermannOp);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_testlib_ops.cc | 39 REGISTER_KERNEL_BUILDER(Name("Error").Device(DEVICE_CPU), ErrorOp);
61 REGISTER_KERNEL_BUILDER(Name("InvalidRefType").Device(DEVICE_CPU),
82 REGISTER_KERNEL_BUILDER(Name("Delay").Device(DEVICE_CPU), DelayOp);
|
