| /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);
106 return {DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}; in DeviceTypes()
171 ExpectSuccess("Test1", DEVICE_CPU, {DT_FLOAT, DT_INT32}, {DT_UINT8}); in TEST_F()
172 ExpectSuccess("Test1", DEVICE_CPU, {DT_FLOAT_REF, DT_INT32}, {DT_UINT8}); in TEST_F()
182 ExpectSuccess("Test3", DEVICE_CPU, {DT_INT8, DT_INT8}, {}); in TEST_F()
191 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0]); in TEST_F()
202 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0]); in TEST_F()
228 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[1]); in TEST_F()
[all …]
|
| D | op_compatibility_test.cc | 199 REGISTER_KERNEL_BUILDER(Name("Same").Device(DEVICE_CPU), TestKernel);
218 REGISTER_KERNEL_BUILDER(Name("AddAttr").Device(DEVICE_CPU), TestKernel);
231 REGISTER_KERNEL_BUILDER(Name("LessStrict").Device(DEVICE_CPU), TestKernel);
248 REGISTER_KERNEL_BUILDER(Name("RemoveRestriction").Device(DEVICE_CPU),
266 REGISTER_KERNEL_BUILDER(Name("AttrOrder").Device(DEVICE_CPU), TestKernel);
289 REGISTER_KERNEL_BUILDER(Name("TypePolymorphic").Device(DEVICE_CPU), TestKernel);
310 REGISTER_KERNEL_BUILDER(Name("MakeList").Device(DEVICE_CPU), TestKernel);
333 REGISTER_KERNEL_BUILDER(Name("MakePolyList").Device(DEVICE_CPU), TestKernel);
354 REGISTER_KERNEL_BUILDER(Name("MakeAnyList").Device(DEVICE_CPU), TestKernel);
376 REGISTER_KERNEL_BUILDER(Name("PolyIntoList").Device(DEVICE_CPU), TestKernel);
[all …]
|
| D | variant_op_registry_test.cc | 101 DEVICE_CPU, VariantValue,
110 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
220 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, "YOU SHALL NOT PASS"), in TEST()
273 registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, kTypeName, in TEST()
276 DEVICE_CPU, kTypeName, f), in TEST()
289 ADD_VARIANT_BINARY_OP, DEVICE_CPU, "YOU SHALL NOT PASS"), in TEST()
344 registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, kTypeName, f); in TEST()
345 EXPECT_DEATH(registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 | 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 …]
|
| 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 | 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"), \
55 Name("Elu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
58 Name("EluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
61 Name("Selu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
64 Name("SeluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
|
| D | summary_kernels.cc | 27 REGISTER_KERNEL_BUILDER(Name("SummaryWriter").Device(DEVICE_CPU),
57 REGISTER_KERNEL_BUILDER(Name("CreateSummaryFileWriter").Device(DEVICE_CPU),
93 REGISTER_KERNEL_BUILDER(Name("CreateSummaryDbWriter").Device(DEVICE_CPU),
107 REGISTER_KERNEL_BUILDER(Name("FlushSummaryWriter").Device(DEVICE_CPU),
119 REGISTER_KERNEL_BUILDER(Name("CloseSummaryWriter").Device(DEVICE_CPU),
144 REGISTER_KERNEL_BUILDER(Name("WriteSummary").Device(DEVICE_CPU),
166 REGISTER_KERNEL_BUILDER(Name("ImportEvent").Device(DEVICE_CPU), ImportEventOp);
188 REGISTER_KERNEL_BUILDER(Name("WriteScalarSummary").Device(DEVICE_CPU),
211 REGISTER_KERNEL_BUILDER(Name("WriteHistogramSummary").Device(DEVICE_CPU),
249 REGISTER_KERNEL_BUILDER(Name("WriteImageSummary").Device(DEVICE_CPU),
[all …]
|
| D | control_flow_ops.cc | 44 .Device(DEVICE_CPU) \
51 .Device(DEVICE_CPU) \
200 .Device(DEVICE_CPU) \
239 REGISTER_KERNEL_BUILDER(Name("Merge").Device(DEVICE_CPU), MergeOp);
240 REGISTER_KERNEL_BUILDER(Name("RefMerge").Device(DEVICE_CPU), MergeOp);
344 REGISTER_KERNEL_BUILDER(Name("Enter").Device(DEVICE_CPU), EnterOp);
345 REGISTER_KERNEL_BUILDER(Name("RefEnter").Device(DEVICE_CPU), EnterOp);
439 REGISTER_KERNEL_BUILDER(Name("Exit").Device(DEVICE_CPU), ExitOp);
440 REGISTER_KERNEL_BUILDER(Name("RefExit").Device(DEVICE_CPU), ExitOp);
518 REGISTER_KERNEL_BUILDER(Name("NextIteration").Device(DEVICE_CPU),
[all …]
|
| D | tensor_array_ops.cc | 137 if (device_type_ == DEVICE_CPU) { in Compute()
230 REGISTER_KERNEL_BUILDER(Name("TensorArray").Device(DEVICE_CPU), TensorArrayOp);
231 REGISTER_KERNEL_BUILDER(Name("TensorArrayV2").Device(DEVICE_CPU),
233 REGISTER_KERNEL_BUILDER(Name("TensorArrayV3").Device(DEVICE_CPU),
358 REGISTER_KERNEL_BUILDER(Name("TensorArrayGrad").Device(DEVICE_CPU),
360 REGISTER_KERNEL_BUILDER(Name("TensorArrayGradV2").Device(DEVICE_CPU),
362 REGISTER_KERNEL_BUILDER(Name("TensorArrayGradV3").Device(DEVICE_CPU),
421 Name("TensorArrayWrite").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
424 Name("TensorArrayWriteV2").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
427 Name("TensorArrayWriteV3").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 | shape_ops.cc | 25 .Device(DEVICE_CPU)
30 .Device(DEVICE_CPU)
112 .Device(DEVICE_CPU)
117 .Device(DEVICE_CPU)
196 REGISTER_KERNEL_BUILDER(Name("Rank").Device(DEVICE_CPU).HostMemory("output"),
256 .Device(DEVICE_CPU)
261 .Device(DEVICE_CPU)
341 .Device(DEVICE_CPU)
346 .Device(DEVICE_CPU)
424 REGISTER_KERNEL_BUILDER(Name("Squeeze").Device(DEVICE_CPU), SqueezeOp);
|
| D | list_kernels.cc | 165 REGISTER_KERNEL_BUILDER(Name("EmptyTensorList").Device(DEVICE_CPU),
223 REGISTER_KERNEL_BUILDER(Name("TensorListPushBack").Device(DEVICE_CPU),
251 REGISTER_KERNEL_BUILDER(Name("TensorListLength").Device(DEVICE_CPU),
289 REGISTER_KERNEL_BUILDER(Name("TensorListElementShape").Device(DEVICE_CPU),
339 REGISTER_KERNEL_BUILDER(Name("TensorListPopBack").Device(DEVICE_CPU),
374 REGISTER_KERNEL_BUILDER(Name("TensorListReserve").Device(DEVICE_CPU),
419 REGISTER_KERNEL_BUILDER(Name("TensorListGetItem").Device(DEVICE_CPU),
466 REGISTER_KERNEL_BUILDER(Name("TensorListSetItem").Device(DEVICE_CPU),
480 .Device(DEVICE_CPU), \
496 .Device(DEVICE_CPU), \
[all …]
|
| D | dequantize_op.cc | 131 Name("Dequantize").Device(DEVICE_CPU).TypeConstraint<quint8>("T"),
134 Name("Dequantize").Device(DEVICE_CPU).TypeConstraint<qint8>("T"),
137 Name("Dequantize").Device(DEVICE_CPU).TypeConstraint<quint16>("T"),
140 Name("Dequantize").Device(DEVICE_CPU).TypeConstraint<qint16>("T"),
144 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 | fft_ops.cc | 251 REGISTER_KERNEL_BUILDER(Name("FFT").Device(DEVICE_CPU).Label(FFT_LABEL),
253 REGISTER_KERNEL_BUILDER(Name("IFFT").Device(DEVICE_CPU).Label(FFT_LABEL),
255 REGISTER_KERNEL_BUILDER(Name("FFT2D").Device(DEVICE_CPU).Label(FFT_LABEL),
257 REGISTER_KERNEL_BUILDER(Name("IFFT2D").Device(DEVICE_CPU).Label(FFT_LABEL),
259 REGISTER_KERNEL_BUILDER(Name("FFT3D").Device(DEVICE_CPU).Label(FFT_LABEL),
261 REGISTER_KERNEL_BUILDER(Name("IFFT3D").Device(DEVICE_CPU).Label(FFT_LABEL),
264 REGISTER_KERNEL_BUILDER(Name("RFFT").Device(DEVICE_CPU).Label(FFT_LABEL),
266 REGISTER_KERNEL_BUILDER(Name("IRFFT").Device(DEVICE_CPU).Label(FFT_LABEL),
268 REGISTER_KERNEL_BUILDER(Name("RFFT2D").Device(DEVICE_CPU).Label(FFT_LABEL),
270 REGISTER_KERNEL_BUILDER(Name("IRFFT2D").Device(DEVICE_CPU).Label(FFT_LABEL),
[all …]
|
| 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),
|
| D | lookup_table_op.cc | 681 REGISTER_KERNEL_BUILDER(Name("LookupTableFind").Device(DEVICE_CPU),
683 REGISTER_KERNEL_BUILDER(Name("LookupTableFindV2").Device(DEVICE_CPU),
718 REGISTER_KERNEL_BUILDER(Name("LookupTableInsert").Device(DEVICE_CPU),
720 REGISTER_KERNEL_BUILDER(Name("LookupTableInsertV2").Device(DEVICE_CPU),
739 REGISTER_KERNEL_BUILDER(Name("LookupTableSize").Device(DEVICE_CPU),
741 REGISTER_KERNEL_BUILDER(Name("LookupTableSizeV2").Device(DEVICE_CPU),
758 REGISTER_KERNEL_BUILDER(Name("LookupTableExport").Device(DEVICE_CPU),
760 REGISTER_KERNEL_BUILDER(Name("LookupTableExportV2").Device(DEVICE_CPU),
795 REGISTER_KERNEL_BUILDER(Name("LookupTableImport").Device(DEVICE_CPU),
797 REGISTER_KERNEL_BUILDER(Name("LookupTableImportV2").Device(DEVICE_CPU),
[all …]
|
| D | in_topk_op.cc | 102 .Device(DEVICE_CPU)
109 .Device(DEVICE_CPU)
117 .Device(DEVICE_CPU)
125 .Device(DEVICE_CPU)
|
| D | map_stage_op.cc | 540 REGISTER_KERNEL_BUILDER(Name("MapStage").Device(DEVICE_CPU), MapStageOp<false>);
541 REGISTER_KERNEL_BUILDER(Name("OrderedMapStage").Device(DEVICE_CPU),
600 REGISTER_KERNEL_BUILDER(Name("MapUnstage").Device(DEVICE_CPU),
602 REGISTER_KERNEL_BUILDER(Name("OrderedMapUnstage").Device(DEVICE_CPU),
662 REGISTER_KERNEL_BUILDER(Name("MapPeek").Device(DEVICE_CPU), MapPeekOp<false>);
663 REGISTER_KERNEL_BUILDER(Name("OrderedMapPeek").Device(DEVICE_CPU),
724 REGISTER_KERNEL_BUILDER(Name("MapUnstageNoKey").Device(DEVICE_CPU),
726 REGISTER_KERNEL_BUILDER(Name("OrderedMapUnstageNoKey").Device(DEVICE_CPU),
774 REGISTER_KERNEL_BUILDER(Name("MapSize").Device(DEVICE_CPU), MapSizeOp<false>);
775 REGISTER_KERNEL_BUILDER(Name("OrderedMapSize").Device(DEVICE_CPU),
[all …]
|
| D | 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),
|
| D | quantized_activation_ops.cc | 99 .Device(DEVICE_CPU)
104 .Device(DEVICE_CPU)
110 .Device(DEVICE_CPU)
115 .Device(DEVICE_CPU)
|
| /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/contrib/boosted_trees/kernels/ |
| D | model_ops.cc | 143 Name("TreeEnsembleIsInitializedOp").Device(DEVICE_CPU),
146 REGISTER_KERNEL_BUILDER(Name("CreateTreeEnsembleVariable").Device(DEVICE_CPU),
149 REGISTER_KERNEL_BUILDER(Name("TreeEnsembleStampToken").Device(DEVICE_CPU),
152 REGISTER_KERNEL_BUILDER(Name("TreeEnsembleSerialize").Device(DEVICE_CPU),
155 REGISTER_KERNEL_BUILDER(Name("TreeEnsembleDeserialize").Device(DEVICE_CPU),
|