| /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 | map_kernels.cc | 26 REGISTER_KERNEL_BUILDER(Name("EmptyTensorMap").Device(DEVICE_CPU),
29 REGISTER_KERNEL_BUILDER(Name("TensorMapSize").Device(DEVICE_CPU),
32 REGISTER_KERNEL_BUILDER(Name("TensorMapLookup").Device(DEVICE_CPU),
35 REGISTER_KERNEL_BUILDER(Name("TensorMapInsert").Device(DEVICE_CPU),
38 REGISTER_KERNEL_BUILDER(Name("TensorMapErase").Device(DEVICE_CPU),
41 REGISTER_KERNEL_BUILDER(Name("TensorMapHasKey").Device(DEVICE_CPU),
44 REGISTER_KERNEL_BUILDER(Name("TensorMapStackKeys").Device(DEVICE_CPU),
51 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
56 DEVICE_CPU, TensorMap,
|
| D | stack_ops.cc | 43 REGISTER_KERNEL_BUILDER(Name("Stack").Device(DEVICE_CPU), StackOp);
47 REGISTER_KERNEL_BUILDER(Name("StackV2").Device(DEVICE_CPU), StackOp);
54 REGISTER_KERNEL_BUILDER(Name("StackPush").Device(DEVICE_CPU),
56 REGISTER_KERNEL_BUILDER(Name("StackPushV2").Device(DEVICE_CPU),
59 REGISTER_KERNEL_BUILDER(Name("StackPop").Device(DEVICE_CPU), StackPopOp);
60 REGISTER_KERNEL_BUILDER(Name("StackPopV2").Device(DEVICE_CPU), StackPopOp);
123 REGISTER_KERNEL_BUILDER(Name("StackClose").Device(DEVICE_CPU), StackCloseOp);
127 REGISTER_KERNEL_BUILDER(Name("StackCloseV2").Device(DEVICE_CPU), StackCloseOp);
|
| D | reader_ops.cc | 102 REGISTER_KERNEL_BUILDER(Name("ReaderRead").Device(DEVICE_CPU), ReaderReadOp);
103 REGISTER_KERNEL_BUILDER(Name("ReaderReadV2").Device(DEVICE_CPU), ReaderReadOp);
154 REGISTER_KERNEL_BUILDER(Name("ReaderReadUpTo").Device(DEVICE_CPU),
156 REGISTER_KERNEL_BUILDER(Name("ReaderReadUpToV2").Device(DEVICE_CPU),
172 REGISTER_KERNEL_BUILDER(Name("ReaderNumRecordsProduced").Device(DEVICE_CPU),
174 REGISTER_KERNEL_BUILDER(Name("ReaderNumRecordsProducedV2").Device(DEVICE_CPU),
190 REGISTER_KERNEL_BUILDER(Name("ReaderNumWorkUnitsCompleted").Device(DEVICE_CPU),
193 Name("ReaderNumWorkUnitsCompletedV2").Device(DEVICE_CPU),
210 REGISTER_KERNEL_BUILDER(Name("ReaderSerializeState").Device(DEVICE_CPU),
212 REGISTER_KERNEL_BUILDER(Name("ReaderSerializeStateV2").Device(DEVICE_CPU),
[all …]
|
| D | segment_reduction_ops_impl_5.cc | 31 .Device(DEVICE_CPU) \
39 .Device(DEVICE_CPU) \
51 .Device(DEVICE_CPU) \
59 .Device(DEVICE_CPU) \
73 .Device(DEVICE_CPU) \
81 .Device(DEVICE_CPU) \
95 .Device(DEVICE_CPU) \
107 .Device(DEVICE_CPU) \
|
| D | debug_ops.cc | 27 REGISTER_KERNEL_BUILDER(Name("Copy").Device(DEVICE_CPU), CopyOp);
29 REGISTER_KERNEL_BUILDER(Name("CopyHost").Device(DEVICE_CPU), CopyOp);
43 REGISTER_KERNEL_BUILDER(Name("DebugIdentity").Device(DEVICE_CPU),
58 Name("DebugNanCount").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
79 .Device(DEVICE_CPU) \
102 REGISTER_KERNEL_BUILDER(Name("DebugIdentityV2").Device(DEVICE_CPU),
115 .Device(DEVICE_CPU) \
129 .Device(DEVICE_CPU) \
|
| D | identity_op.cc | 26 REGISTER_KERNEL_BUILDER(Name("Identity").Device(DEVICE_CPU), IdentityOp);
31 REGISTER_KERNEL_BUILDER(Name("StopGradient").Device(DEVICE_CPU), IdentityOp);
34 REGISTER_KERNEL_BUILDER(Name("PreventGradient").Device(DEVICE_CPU), IdentityOp);
38 REGISTER_KERNEL_BUILDER(Name("PlaceholderWithDefault").Device(DEVICE_CPU),
41 REGISTER_KERNEL_BUILDER(Name("RefIdentity").Device(DEVICE_CPU), IdentityOp);
45 REGISTER_KERNEL_BUILDER(Name("DebugGradientIdentity").Device(DEVICE_CPU),
47 REGISTER_KERNEL_BUILDER(Name("DebugGradientRefIdentity").Device(DEVICE_CPU),
|
| D | shape_ops.cc | 26 .Device(DEVICE_CPU)
31 .Device(DEVICE_CPU)
79 .Device(DEVICE_CPU)
84 .Device(DEVICE_CPU)
129 REGISTER_KERNEL_BUILDER(Name("Rank").Device(DEVICE_CPU).HostMemory("output"),
165 .Device(DEVICE_CPU)
170 .Device(DEVICE_CPU)
217 .Device(DEVICE_CPU)
222 .Device(DEVICE_CPU)
265 REGISTER_KERNEL_BUILDER(Name("Squeeze").Device(DEVICE_CPU), SqueezeOp);
[all …]
|
| D | list_kernels.cc | 159 REGISTER_KERNEL_BUILDER(Name("EmptyTensorList").Device(DEVICE_CPU),
219 REGISTER_KERNEL_BUILDER(Name("TensorListPushBack").Device(DEVICE_CPU),
243 REGISTER_KERNEL_BUILDER(Name("TensorListLength").Device(DEVICE_CPU),
283 REGISTER_KERNEL_BUILDER(Name("TensorListElementShape").Device(DEVICE_CPU),
320 REGISTER_KERNEL_BUILDER(Name("TensorListReserve").Device(DEVICE_CPU),
383 REGISTER_KERNEL_BUILDER(Name("TensorListResize").Device(DEVICE_CPU),
429 REGISTER_KERNEL_BUILDER(Name("TensorListSetItem").Device(DEVICE_CPU),
544 REGISTER_KERNEL_BUILDER(Name("TensorListConcatLists").Device(DEVICE_CPU),
557 .Device(DEVICE_CPU), \
561 .Device(DEVICE_CPU), \
[all …]
|
| D | relu_op.cc | 36 Name("Relu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
39 Name("ReluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
42 Name("Relu6").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
45 Name("Relu6Grad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
48 Name("LeakyRelu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
51 Name("LeakyReluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
59 Name("Elu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
62 Name("EluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
65 Name("Selu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
68 Name("SeluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
|
| D | dequantize_op.cc | 231 .Device(DEVICE_CPU)
236 .Device(DEVICE_CPU)
241 .Device(DEVICE_CPU)
246 .Device(DEVICE_CPU)
251 .Device(DEVICE_CPU)
257 .Device(DEVICE_CPU)
262 .Device(DEVICE_CPU)
267 .Device(DEVICE_CPU)
272 .Device(DEVICE_CPU)
277 .Device(DEVICE_CPU)
|
| D | summary_kernels.cc | 30 REGISTER_KERNEL_BUILDER(Name("SummaryWriter").Device(DEVICE_CPU),
60 REGISTER_KERNEL_BUILDER(Name("CreateSummaryFileWriter").Device(DEVICE_CPU),
96 REGISTER_KERNEL_BUILDER(Name("CreateSummaryDbWriter").Device(DEVICE_CPU),
109 REGISTER_KERNEL_BUILDER(Name("FlushSummaryWriter").Device(DEVICE_CPU),
121 REGISTER_KERNEL_BUILDER(Name("CloseSummaryWriter").Device(DEVICE_CPU),
145 REGISTER_KERNEL_BUILDER(Name("WriteSummary").Device(DEVICE_CPU),
181 REGISTER_KERNEL_BUILDER(Name("WriteRawProtoSummary").Device(DEVICE_CPU),
202 REGISTER_KERNEL_BUILDER(Name("ImportEvent").Device(DEVICE_CPU), ImportEventOp);
223 REGISTER_KERNEL_BUILDER(Name("WriteScalarSummary").Device(DEVICE_CPU),
245 REGISTER_KERNEL_BUILDER(Name("WriteHistogramSummary").Device(DEVICE_CPU),
[all …]
|
| D | tensor_array_ops.cc | 139 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),
380 REGISTER_KERNEL_BUILDER(Name("TensorArrayGrad").Device(DEVICE_CPU),
382 REGISTER_KERNEL_BUILDER(Name("TensorArrayGradV2").Device(DEVICE_CPU),
384 REGISTER_KERNEL_BUILDER(Name("TensorArrayGradV3").Device(DEVICE_CPU),
386 REGISTER_KERNEL_BUILDER(Name("TensorArrayGradWithShape").Device(DEVICE_CPU),
450 Name("TensorArrayWrite").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
453 Name("TensorArrayWriteV2").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
[all …]
|
| D | control_flow_ops.cc | 66 .Device(DEVICE_CPU) \
71 .Device(DEVICE_CPU) \
78 .Device(DEVICE_CPU) \
193 .Device(DEVICE_CPU) \
236 REGISTER_KERNEL_BUILDER(Name("Merge").Device(DEVICE_CPU), MergeOp);
241 REGISTER_KERNEL_BUILDER(Name("RefMerge").Device(DEVICE_CPU), MergeOp);
305 REGISTER_KERNEL_BUILDER(Name("RefEnter").Device(DEVICE_CPU), EnterOp);
362 REGISTER_KERNEL_BUILDER(Name("RefExit").Device(DEVICE_CPU), ExitOp);
416 REGISTER_KERNEL_BUILDER(Name("RefNextIteration").Device(DEVICE_CPU),
473 REGISTER_KERNEL_BUILDER(Name("LoopCond").Device(DEVICE_CPU), LoopCondOp);
[all …]
|
| D | conditional_accumulator_base_op.cc | 61 REGISTER_KERNEL_BUILDER(Name("AccumulatorSetGlobalStep").Device(DEVICE_CPU),
79 Name("ResourceAccumulatorSetGlobalStep").Device(DEVICE_CPU),
122 REGISTER_KERNEL_BUILDER(Name("AccumulatorNumAccumulated").Device(DEVICE_CPU),
140 Name("ResourceAccumulatorNumAccumulated").Device(DEVICE_CPU),
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 52 REGISTER_KERNEL_BUILDER(Name("Test1").Device(tensorflow::DEVICE_CPU),
91 Name("Test3").Device(DEVICE_CPU).TypeConstraint<int8>("T"), TestOp3Cpu);
106 REGISTER_KERNEL_BUILDER(Name("Test4").Device(DEVICE_CPU), DummyKernel);
120 REGISTER_KERNEL_BUILDER(Name("Test5").Device(DEVICE_CPU).Priority(2),
133 return {DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}; in DeviceTypes()
200 ExpectSuccess("Test1", DEVICE_CPU, {DT_FLOAT, DT_INT32}, {DT_UINT8}); in TEST_F()
201 ExpectSuccess("Test1", DEVICE_CPU, {DT_FLOAT_REF, DT_INT32}, {DT_UINT8}); in TEST_F()
211 ExpectSuccess("Test3", DEVICE_CPU, {DT_INT8, DT_INT8}, {}); in TEST_F()
220 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0].first); in TEST_F()
238 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[1].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 …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_tmp_bf16_ops.cc | 35 Name("Conv2D").Device(DEVICE_CPU).TypeConstraint<T>("T"), NoOp); \
37 Name("Conv2DBackpropFilter").Device(DEVICE_CPU).TypeConstraint<T>("T"), \
40 Name("Conv2DBackpropInput").Device(DEVICE_CPU).TypeConstraint<T>("T"), \
43 Name("_FusedConv2D").Device(DEVICE_CPU).TypeConstraint<T>("T"), NoOp); \
45 Name("AvgPool").Device(DEVICE_CPU).TypeConstraint<T>("T"), NoOp); \
47 Name("AvgPoolGrad").Device(DEVICE_CPU).TypeConstraint<T>("T"), NoOp); \
49 .Device(DEVICE_CPU) \
54 .Device(DEVICE_CPU) \
59 Name("_FusedMatMul").Device(DEVICE_CPU).TypeConstraint<T>("T"), NoOp);
|
| D | mkl_conv_ops.cc | 1842 .Device(DEVICE_CPU)
1849 .Device(DEVICE_CPU)
1857 .Device(DEVICE_CPU)
1864 .Device(DEVICE_CPU)
1874 .Device(DEVICE_CPU)
1883 .Device(DEVICE_CPU)
1891 .Device(DEVICE_CPU)
1903 .Device(DEVICE_CPU)
1910 .Device(DEVICE_CPU)
1917 .Device(DEVICE_CPU)
[all …]
|
| D | mkl_avgpooling_op.cc | 302 .Device(DEVICE_CPU) \
308 .Device(DEVICE_CPU) \
313 .Device(DEVICE_CPU) \
318 .Device(DEVICE_CPU) \
329 .Device(DEVICE_CPU) \
335 .Device(DEVICE_CPU) \
340 .Device(DEVICE_CPU) \
345 .Device(DEVICE_CPU) \
354 .Device(DEVICE_CPU)
360 .Device(DEVICE_CPU)
|
| D | mkl_qmatmul_op.cc | 535 .Device(DEVICE_CPU)
543 .Device(DEVICE_CPU)
552 .Device(DEVICE_CPU)
564 .Device(DEVICE_CPU)
573 .Device(DEVICE_CPU)
584 .Device(DEVICE_CPU)
595 .Device(DEVICE_CPU)
605 .Device(DEVICE_CPU)
615 .Device(DEVICE_CPU)
624 .Device(DEVICE_CPU)
[all …]
|
| /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/core/kernels/data/ |
| D | random_seed_ops.cc | 106 REGISTER_KERNEL_BUILDER(Name("AnonymousSeedGenerator").Device(DEVICE_CPU),
109 REGISTER_KERNEL_BUILDER(Name("DeleteSeedGenerator").Device(DEVICE_CPU),
112 REGISTER_KERNEL_BUILDER(Name("AnonymousRandomSeedGenerator").Device(DEVICE_CPU),
115 REGISTER_KERNEL_BUILDER(Name("DeleteRandomSeedGenerator").Device(DEVICE_CPU),
118 REGISTER_KERNEL_BUILDER(Name("DummySeedGenerator").Device(DEVICE_CPU),
|
| /external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
| D | resource_ops.cc | 117 Name("TensorForestTreeIsInitializedOp").Device(DEVICE_CPU),
121 Name("TensorForestCreateTreeVariable").Device(DEVICE_CPU),
124 REGISTER_KERNEL_BUILDER(Name("TensorForestTreeSerialize").Device(DEVICE_CPU),
127 REGISTER_KERNEL_BUILDER(Name("TensorForestTreeDeserialize").Device(DEVICE_CPU),
130 REGISTER_KERNEL_BUILDER(Name("TensorForestTreeSize").Device(DEVICE_CPU),
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_ops.cc | 117 REGISTER_KERNEL_BUILDER(Name("KernelLabel").Device(DEVICE_CPU),
120 .Device(DEVICE_CPU)
124 .Device(DEVICE_CPU)
130 Name("KernelLabelRequired").Device(DEVICE_CPU).Label("overload_1"),
133 Name("KernelLabelRequired").Device(DEVICE_CPU).Label("overload_2"),
151 REGISTER_KERNEL_BUILDER(Name("GraphDefVersion").Device(DEVICE_CPU),
161 REGISTER_KERNEL_BUILDER(Name("Old").Device(DEVICE_CPU), OldOp);
171 REGISTER_KERNEL_BUILDER(Name("ResourceInitializedOp").Device(DEVICE_CPU),
184 REGISTER_KERNEL_BUILDER(Name("ResourceCreateOp").Device(DEVICE_CPU),
199 REGISTER_KERNEL_BUILDER(Name("ResourceUsingOp").Device(DEVICE_CPU),
[all …]
|