| /external/tensorflow/tensorflow/core/kernels/ |
| D | queue_ops.cc | 27 REGISTER_KERNEL_BUILDER(Name("QueueEnqueue").Device(DEVICE_CPU), EnqueueOp);
28 REGISTER_KERNEL_BUILDER(Name("QueueEnqueueV2").Device(DEVICE_CPU), EnqueueOp);
30 REGISTER_KERNEL_BUILDER(Name("QueueEnqueueMany").Device(DEVICE_CPU),
32 REGISTER_KERNEL_BUILDER(Name("QueueEnqueueManyV2").Device(DEVICE_CPU),
35 REGISTER_KERNEL_BUILDER(Name("QueueDequeue").Device(DEVICE_CPU), DequeueOp);
36 REGISTER_KERNEL_BUILDER(Name("QueueDequeueV2").Device(DEVICE_CPU), DequeueOp);
38 REGISTER_KERNEL_BUILDER(Name("QueueDequeueMany").Device(DEVICE_CPU),
40 REGISTER_KERNEL_BUILDER(Name("QueueDequeueManyV2").Device(DEVICE_CPU),
43 REGISTER_KERNEL_BUILDER(Name("QueueDequeueUpTo").Device(DEVICE_CPU),
45 REGISTER_KERNEL_BUILDER(Name("QueueDequeueUpToV2").Device(DEVICE_CPU),
[all …]
|
| D | 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 …]
|
| D | stack_ops.cc | 43 REGISTER_KERNEL_BUILDER(Name("Stack").Device(DEVICE_CPU), StackOp);
46 REGISTER_KERNEL_BUILDER(Name("StackV2").Device(DEVICE_CPU), StackOp);
62 REGISTER_KERNEL_BUILDER(Name("StackPush").Device(DEVICE_CPU),
64 REGISTER_KERNEL_BUILDER(Name("StackPushV2").Device(DEVICE_CPU),
131 REGISTER_KERNEL_BUILDER(Name("StackPop").Device(DEVICE_CPU), StackPopOp);
132 REGISTER_KERNEL_BUILDER(Name("StackPopV2").Device(DEVICE_CPU), StackPopOp);
196 REGISTER_KERNEL_BUILDER(Name("StackClose").Device(DEVICE_CPU), StackCloseOp);
199 REGISTER_KERNEL_BUILDER(Name("StackCloseV2").Device(DEVICE_CPU), StackCloseOp);
|
| D | 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),
|
| D | list_kernels.cc | 251 REGISTER_KERNEL_BUILDER(Name("EmptyTensorList").Device(DEVICE_CPU),
311 REGISTER_KERNEL_BUILDER(Name("TensorListPushBack").Device(DEVICE_CPU),
335 REGISTER_KERNEL_BUILDER(Name("TensorListLength").Device(DEVICE_CPU),
375 REGISTER_KERNEL_BUILDER(Name("TensorListElementShape").Device(DEVICE_CPU),
403 REGISTER_KERNEL_BUILDER(Name("TensorListReserve").Device(DEVICE_CPU),
459 REGISTER_KERNEL_BUILDER(Name("TensorListResize").Device(DEVICE_CPU),
505 REGISTER_KERNEL_BUILDER(Name("TensorListSetItem").Device(DEVICE_CPU),
610 REGISTER_KERNEL_BUILDER(Name("TensorListConcatLists").Device(DEVICE_CPU),
623 .Device(DEVICE_CPU), \
627 .Device(DEVICE_CPU), \
[all …]
|
| D | relu_op.cc | 38 Name("Relu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
41 Name("ReluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
44 Name("Relu6").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
47 Name("Relu6Grad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
50 Name("LeakyRelu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
53 Name("LeakyReluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
61 Name("Elu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
64 Name("EluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
67 Name("Selu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
70 Name("SeluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
|
| D | summary_kernels.cc | 28 REGISTER_KERNEL_BUILDER(Name("SummaryWriter").Device(DEVICE_CPU),
59 REGISTER_KERNEL_BUILDER(Name("CreateSummaryFileWriter").Device(DEVICE_CPU),
96 REGISTER_KERNEL_BUILDER(Name("CreateSummaryDbWriter").Device(DEVICE_CPU),
110 REGISTER_KERNEL_BUILDER(Name("FlushSummaryWriter").Device(DEVICE_CPU),
122 REGISTER_KERNEL_BUILDER(Name("CloseSummaryWriter").Device(DEVICE_CPU),
147 REGISTER_KERNEL_BUILDER(Name("WriteSummary").Device(DEVICE_CPU),
169 REGISTER_KERNEL_BUILDER(Name("ImportEvent").Device(DEVICE_CPU), ImportEventOp);
191 REGISTER_KERNEL_BUILDER(Name("WriteScalarSummary").Device(DEVICE_CPU),
214 REGISTER_KERNEL_BUILDER(Name("WriteHistogramSummary").Device(DEVICE_CPU),
252 REGISTER_KERNEL_BUILDER(Name("WriteImageSummary").Device(DEVICE_CPU),
[all …]
|
| D | tensor_array_ops.cc | 138 if (device_type_ == DEVICE_CPU) { in Compute()
231 REGISTER_KERNEL_BUILDER(Name("TensorArray").Device(DEVICE_CPU), TensorArrayOp);
232 REGISTER_KERNEL_BUILDER(Name("TensorArrayV2").Device(DEVICE_CPU),
234 REGISTER_KERNEL_BUILDER(Name("TensorArrayV3").Device(DEVICE_CPU),
383 REGISTER_KERNEL_BUILDER(Name("TensorArrayGrad").Device(DEVICE_CPU),
385 REGISTER_KERNEL_BUILDER(Name("TensorArrayGradV2").Device(DEVICE_CPU),
387 REGISTER_KERNEL_BUILDER(Name("TensorArrayGradV3").Device(DEVICE_CPU),
389 REGISTER_KERNEL_BUILDER(Name("TensorArrayGradWithShape").Device(DEVICE_CPU),
453 Name("TensorArrayWrite").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
456 Name("TensorArrayWriteV2").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
[all …]
|
| D | 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) \
|
| D | control_flow_ops.cc | 44 .Device(DEVICE_CPU) \
51 .Device(DEVICE_CPU) \
207 .Device(DEVICE_CPU) \
246 REGISTER_KERNEL_BUILDER(Name("Merge").Device(DEVICE_CPU), MergeOp);
247 REGISTER_KERNEL_BUILDER(Name("RefMerge").Device(DEVICE_CPU), MergeOp);
355 REGISTER_KERNEL_BUILDER(Name("Enter").Device(DEVICE_CPU), EnterOp);
356 REGISTER_KERNEL_BUILDER(Name("RefEnter").Device(DEVICE_CPU), EnterOp);
451 REGISTER_KERNEL_BUILDER(Name("Exit").Device(DEVICE_CPU), ExitOp);
452 REGISTER_KERNEL_BUILDER(Name("RefExit").Device(DEVICE_CPU), ExitOp);
532 REGISTER_KERNEL_BUILDER(Name("NextIteration").Device(DEVICE_CPU),
[all …]
|
| D | shape_ops.cc | 26 .Device(DEVICE_CPU)
31 .Device(DEVICE_CPU)
113 .Device(DEVICE_CPU)
118 .Device(DEVICE_CPU)
197 REGISTER_KERNEL_BUILDER(Name("Rank").Device(DEVICE_CPU).HostMemory("output"),
257 .Device(DEVICE_CPU)
262 .Device(DEVICE_CPU)
342 .Device(DEVICE_CPU)
347 .Device(DEVICE_CPU)
425 REGISTER_KERNEL_BUILDER(Name("Squeeze").Device(DEVICE_CPU), SqueezeOp);
[all …]
|
| D | dequantize_op.cc | 119 Name("Dequantize").Device(DEVICE_CPU).TypeConstraint<quint8>("T"),
122 Name("Dequantize").Device(DEVICE_CPU).TypeConstraint<qint8>("T"),
125 Name("Dequantize").Device(DEVICE_CPU).TypeConstraint<quint16>("T"),
128 Name("Dequantize").Device(DEVICE_CPU).TypeConstraint<qint16>("T"),
132 Name("Dequantize").Device(DEVICE_CPU).TypeConstraint<qint32>("T"),
|
| D | save_op.cc | 38 REGISTER_KERNEL_BUILDER(Name("Save").Device(DEVICE_CPU), SaveOp);
49 REGISTER_KERNEL_BUILDER(Name("SaveSlices").Device(DEVICE_CPU), SaveSlicesOp);
71 REGISTER_KERNEL_BUILDER(Name("ShardedFilename").Device(DEVICE_CPU),
93 REGISTER_KERNEL_BUILDER(Name("ShardedFilespec").Device(DEVICE_CPU),
|
| D | mkl_conv_ops.cc | 841 .Device(DEVICE_CPU)
846 .Device(DEVICE_CPU)
1848 .Device(DEVICE_CPU)
1855 .Device(DEVICE_CPU)
1864 .Device(DEVICE_CPU)
1873 .Device(DEVICE_CPU)
1884 .Device(DEVICE_CPU)
1891 .Device(DEVICE_CPU)
1900 .Device(DEVICE_CPU)
1909 .Device(DEVICE_CPU)
[all …]
|
| D | unique_op.cc | 199 .Device(DEVICE_CPU) \
204 .Device(DEVICE_CPU) \
209 .Device(DEVICE_CPU) \
214 .Device(DEVICE_CPU) \
219 .Device(DEVICE_CPU) \
224 .Device(DEVICE_CPU) \
229 .Device(DEVICE_CPU) \
234 .Device(DEVICE_CPU) \
|
| D | candidate_sampler_ops.cc | 123 REGISTER_KERNEL_BUILDER(Name("UniformCandidateSampler").Device(DEVICE_CPU),
126 REGISTER_KERNEL_BUILDER(Name("LogUniformCandidateSampler").Device(DEVICE_CPU),
130 Name("LearnedUnigramCandidateSampler").Device(DEVICE_CPU),
134 Name("ThreadUnsafeUnigramCandidateSampler").Device(DEVICE_CPU),
147 REGISTER_KERNEL_BUILDER(Name("AllCandidateSampler").Device(DEVICE_CPU),
187 REGISTER_KERNEL_BUILDER(Name("FixedUnigramCandidateSampler").Device(DEVICE_CPU),
263 REGISTER_KERNEL_BUILDER(Name("ComputeAccidentalHits").Device(DEVICE_CPU),
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | 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);
114 REGISTER_KERNEL_BUILDER(Name("Test5").Device(DEVICE_CPU).Priority(2),
127 return {DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}; in DeviceTypes()
192 ExpectSuccess("Test1", DEVICE_CPU, {DT_FLOAT, DT_INT32}, {DT_UINT8}); in TEST_F()
193 ExpectSuccess("Test1", DEVICE_CPU, {DT_FLOAT_REF, DT_INT32}, {DT_UINT8}); in TEST_F()
203 ExpectSuccess("Test3", DEVICE_CPU, {DT_INT8, DT_INT8}, {}); in TEST_F()
212 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0].first); in TEST_F()
223 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0].first); in TEST_F()
[all …]
|
| D | op_compatibility_test.cc | 200 REGISTER_KERNEL_BUILDER(Name("Same").Device(DEVICE_CPU), TestKernel);
219 REGISTER_KERNEL_BUILDER(Name("AddAttr").Device(DEVICE_CPU), TestKernel);
232 REGISTER_KERNEL_BUILDER(Name("LessStrict").Device(DEVICE_CPU), TestKernel);
249 REGISTER_KERNEL_BUILDER(Name("RemoveRestriction").Device(DEVICE_CPU),
268 REGISTER_KERNEL_BUILDER(Name("AttrOrder").Device(DEVICE_CPU), TestKernel);
291 REGISTER_KERNEL_BUILDER(Name("TypePolymorphic").Device(DEVICE_CPU), TestKernel);
313 REGISTER_KERNEL_BUILDER(Name("MakeList").Device(DEVICE_CPU), TestKernel);
336 REGISTER_KERNEL_BUILDER(Name("MakePolyList").Device(DEVICE_CPU), TestKernel);
358 REGISTER_KERNEL_BUILDER(Name("MakeAnyList").Device(DEVICE_CPU), TestKernel);
380 REGISTER_KERNEL_BUILDER(Name("PolyIntoList").Device(DEVICE_CPU), TestKernel);
[all …]
|
| D | variant_op_registry_test.cc | 92 DEVICE_CPU, VariantValue,
99 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
180 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, MakeTypeIndex<Blah>()), in TEST()
235 registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, in TEST()
238 DEVICE_CPU, kTypeIndex, f), in TEST()
251 ADD_VARIANT_BINARY_OP, DEVICE_CPU, MakeTypeIndex<Blah>()), in TEST()
308 registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, kTypeIndex, f); in TEST()
309 EXPECT_DEATH(registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, in TEST()
|
| /external/tensorflow/tensorflow/examples/adding_an_op/ |
| D | 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/framework/ |
| D | test_ops.cc | 111 REGISTER_KERNEL_BUILDER(Name("KernelLabel").Device(DEVICE_CPU),
114 .Device(DEVICE_CPU)
118 .Device(DEVICE_CPU)
124 Name("KernelLabelRequired").Device(DEVICE_CPU).Label("overload_1"),
127 Name("KernelLabelRequired").Device(DEVICE_CPU).Label("overload_2"),
145 REGISTER_KERNEL_BUILDER(Name("GraphDefVersion").Device(DEVICE_CPU),
155 REGISTER_KERNEL_BUILDER(Name("Old").Device(DEVICE_CPU), OldOp);
165 REGISTER_KERNEL_BUILDER(Name("ResourceInitializedOp").Device(DEVICE_CPU),
178 REGISTER_KERNEL_BUILDER(Name("ResourceCreateOp").Device(DEVICE_CPU),
193 REGISTER_KERNEL_BUILDER(Name("ResourceUsingOp").Device(DEVICE_CPU),
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
| D | resource_ops.cc | 120 Name("TensorForestTreeIsInitializedOp").Device(DEVICE_CPU),
124 Name("TensorForestCreateTreeVariable").Device(DEVICE_CPU),
127 REGISTER_KERNEL_BUILDER(Name("TensorForestTreeSerialize").Device(DEVICE_CPU),
130 REGISTER_KERNEL_BUILDER(Name("TensorForestTreeDeserialize").Device(DEVICE_CPU),
133 REGISTER_KERNEL_BUILDER(Name("TensorForestTreeSize").Device(DEVICE_CPU),
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 160 REGISTER_KERNEL_BUILDER(Name("OptionalNone").Device(DEVICE_CPU).Priority(2),
165 Name("OptionalFromValue").Device(DEVICE_CPU).Priority(2),
171 REGISTER_KERNEL_BUILDER(Name("OptionalHasValue").Device(DEVICE_CPU).Priority(2),
178 REGISTER_KERNEL_BUILDER(Name("OptionalGetValue").Device(DEVICE_CPU).Priority(2),
186 DEVICE_CPU, OptionalVariant,
189 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | model_ops.cc | 198 Name("TreeEnsembleIsInitializedOp").Device(DEVICE_CPU),
201 REGISTER_KERNEL_BUILDER(Name("CreateTreeEnsembleVariable").Device(DEVICE_CPU),
204 REGISTER_KERNEL_BUILDER(Name("TreeEnsembleStampToken").Device(DEVICE_CPU),
207 REGISTER_KERNEL_BUILDER(Name("TreeEnsembleSerialize").Device(DEVICE_CPU),
210 REGISTER_KERNEL_BUILDER(Name("TreeEnsembleDeserialize").Device(DEVICE_CPU),
213 REGISTER_KERNEL_BUILDER(Name("TreeEnsembleUsedHandlers").Device(DEVICE_CPU),
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | resource_ops.cc | 31 Name("IsBoostedTreesEnsembleInitialized").Device(DEVICE_CPU),
70 REGISTER_KERNEL_BUILDER(Name("BoostedTreesCreateEnsemble").Device(DEVICE_CPU),
134 Name("BoostedTreesGetEnsembleStates").Device(DEVICE_CPU),
162 Name("BoostedTreesSerializeEnsemble").Device(DEVICE_CPU),
198 Name("BoostedTreesDeserializeEnsemble").Device(DEVICE_CPU),
|