Home
last modified time | relevance | path

Searched refs:DEVICE_GPU (Results 1 – 25 of 254) sorted by relevance

1234567891011

/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.cu.cc42 .Device(DEVICE_GPU) \
47 .Device(DEVICE_GPU) \
53 .Device(DEVICE_GPU) \
59 .Device(DEVICE_GPU) \
64 .Device(DEVICE_GPU) \
69 .Device(DEVICE_GPU) \
76 .Device(DEVICE_GPU), \
80 .Device(DEVICE_GPU) \
85 .Device(DEVICE_GPU) \
91 .Device(DEVICE_GPU) \
[all …]
Dshape_ops.cc40 .Device(DEVICE_GPU) \
46 .Device(DEVICE_GPU) \
61 .Device(DEVICE_GPU)
68 .Device(DEVICE_GPU)
92 .Device(DEVICE_GPU) \
98 .Device(DEVICE_GPU) \
112 .Device(DEVICE_GPU)
119 .Device(DEVICE_GPU)
136 .Device(DEVICE_GPU) \
148 .Device(DEVICE_GPU)
[all …]
Didentity_op.cc66 Name("Identity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
69 Name("PreventGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
72 Name("RefIdentity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
75 Name("StopGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
78 .Device(DEVICE_GPU) \
82 .Device(DEVICE_GPU) \
99 .Device(DEVICE_GPU) \
105 .Device(DEVICE_GPU) \
111 .Device(DEVICE_GPU) \
117 .Device(DEVICE_GPU) \
Dcontrol_flow_ops.cc85 .Device(DEVICE_GPU) \
90 .Device(DEVICE_GPU) \
97 .Device(DEVICE_GPU) \
125 .Device(DEVICE_GPU) \
133 .Device(DEVICE_GPU) \
142 .Device(DEVICE_GPU) \
245 .Device(DEVICE_GPU) \
252 .Device(DEVICE_GPU) \
274 .Device(DEVICE_GPU) \
281 .Device(DEVICE_GPU) \
[all …]
Drelu_op.cc156 Name("Relu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
159 Name("ReluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
162 Name("Relu6").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
165 Name("Relu6Grad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
168 Name("LeakyRelu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
171 Name("LeakyReluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
174 Name("Elu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
177 Name("EluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
180 Name("Selu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
183 Name("SeluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
[all …]
Ddebug_ops.cc32 REGISTER_KERNEL_BUILDER(Name("Copy").Device(DEVICE_GPU), CopyOp);
35 .Device(DEVICE_GPU)
48 .Device(DEVICE_GPU)
65 .Device(DEVICE_GPU) \
90 .Device(DEVICE_GPU) \
107 .Device(DEVICE_GPU)
145 .Device(DEVICE_GPU) \
Dfunction_ops.cc100 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), ArgOp);
104 .Device(DEVICE_GPU)
109 Name(kDeviceArgOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), ArgOp);
113 .Device(DEVICE_GPU)
119 .Device(DEVICE_GPU)
125 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<Variant>("T"), ArgOp);
129 Name(kRetOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), RetvalOp);
134 .Device(DEVICE_GPU)
139 Name(kDeviceRetOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), RetvalOp);
142 .Device(DEVICE_GPU)
[all …]
Dtensor_array_ops.cc241 .Device(DEVICE_GPU) \
247 .Device(DEVICE_GPU) \
253 .Device(DEVICE_GPU) \
389 .Device(DEVICE_GPU)
394 .Device(DEVICE_GPU)
399 .Device(DEVICE_GPU)
404 .Device(DEVICE_GPU)
467 .Device(DEVICE_GPU) \
473 .Device(DEVICE_GPU) \
479 .Device(DEVICE_GPU) \
[all …]
Dreshape_op.cc34 .Device(DEVICE_GPU) \
40 .Device(DEVICE_GPU) \
56 .Device(DEVICE_GPU)
64 .Device(DEVICE_GPU)
Dfft_ops.cc574 REGISTER_KERNEL_BUILDER(Name("FFT").Device(DEVICE_GPU).Priority(1),
576 REGISTER_KERNEL_BUILDER(Name("IFFT").Device(DEVICE_GPU).Priority(1),
578 REGISTER_KERNEL_BUILDER(Name("FFT2D").Device(DEVICE_GPU).Priority(1),
580 REGISTER_KERNEL_BUILDER(Name("IFFT2D").Device(DEVICE_GPU).Priority(1),
582 REGISTER_KERNEL_BUILDER(Name("FFT3D").Device(DEVICE_GPU).Priority(1),
584 REGISTER_KERNEL_BUILDER(Name("IFFT3D").Device(DEVICE_GPU).Priority(1),
588 Name("RFFT").Device(DEVICE_GPU).HostMemory("fft_length").Priority(1),
591 Name("IRFFT").Device(DEVICE_GPU).HostMemory("fft_length").Priority(1),
594 Name("RFFT2D").Device(DEVICE_GPU).HostMemory("fft_length").Priority(1),
597 Name("IRFFT2D").Device(DEVICE_GPU).HostMemory("fft_length").Priority(1),
[all …]
Dreduction_ops_sum.cc41 .Device(DEVICE_GPU) \
48 .Device(DEVICE_GPU) \
65 .Device(DEVICE_GPU)
74 .Device(DEVICE_GPU)
Dreduction_ops_min.cc43 .Device(DEVICE_GPU) \
51 .Device(DEVICE_GPU) \
66 .Device(DEVICE_GPU)
75 .Device(DEVICE_GPU)
Dreduction_ops_max.cc43 .Device(DEVICE_GPU) \
51 .Device(DEVICE_GPU) \
68 .Device(DEVICE_GPU)
77 .Device(DEVICE_GPU)
Dcwise_op_conj.cc27 Name("Conj").Device(DEVICE_GPU).TypeConstraint<Variant>("T"),
31 Name("Conj").Device(DEVICE_GPU).TypeConstraint<complex64>("T"),
34 Name("Conj").Device(DEVICE_GPU).TypeConstraint<complex128>("T"),
Dinplace_ops.cc165 .Device(DEVICE_GPU) \
173 Name("ParallelConcat").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
180 .Device(DEVICE_GPU) \
190 .Device(DEVICE_GPU)
440 Name("InplaceUpdate").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \
443 Name("InplaceAdd").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \
446 Name("InplaceSub").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \
449 Name("DeepCopy").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \
453 Name("InplaceUpdate").Device(DEVICE_GPU).TypeConstraint<bool>("T"),
461 .Device(DEVICE_GPU)
[all …]
Dnccl_ops.cc122 REGISTER_KERNEL_BUILDER(Name("NcclAllReduce").Device(DEVICE_GPU),
153 REGISTER_KERNEL_BUILDER(Name("_NcclReduceSend").Device(DEVICE_GPU),
192 REGISTER_KERNEL_BUILDER(Name("_NcclReduceRecv").Device(DEVICE_GPU),
221 REGISTER_KERNEL_BUILDER(Name("_NcclBroadcastSend").Device(DEVICE_GPU),
258 Name("_NcclBroadcastRecv").Device(DEVICE_GPU).HostMemory("shape"),
271 REGISTER_KERNEL_BUILDER(Name("NcclBroadcast").Device(DEVICE_GPU),
273 REGISTER_KERNEL_BUILDER(Name("NcclReduce").Device(DEVICE_GPU), NcclStubKernel);
Dcwise_op_equal_to_1.cc38 Name("ApproximateEqual").Device(DEVICE_GPU).TypeConstraint<float>("T"),
41 Name("ApproximateEqual").Device(DEVICE_GPU).TypeConstraint<double>("T"),
48 .Device(DEVICE_GPU)
Dscan_ops.cc141 .Device(DEVICE_GPU) \
148 .Device(DEVICE_GPU) \
180 .Device(DEVICE_GPU) \
187 .Device(DEVICE_GPU) \
216 REGISTER_CUMLOGSUMEXP_KERNEL(DEVICE_GPU, GPUDevice, type, int32) \
217 REGISTER_CUMLOGSUMEXP_KERNEL(DEVICE_GPU, GPUDevice, type, int64)
Dsession_ops.cc80 .Device(DEVICE_GPU) \
85 .Device(DEVICE_GPU) \
116 .Device(DEVICE_GPU) \
143 Name("DeleteSessionTensor").Device(DEVICE_GPU).HostMemory("handle"),
Dmap_stage_op.cc550 Name("MapStage").HostMemory("key").HostMemory("indices").Device(DEVICE_GPU),
555 .Device(DEVICE_GPU),
600 .Device(DEVICE_GPU),
605 .Device(DEVICE_GPU),
646 Name("MapPeek").HostMemory("key").HostMemory("indices").Device(DEVICE_GPU),
651 .Device(DEVICE_GPU),
701 .Device(DEVICE_GPU),
706 .Device(DEVICE_GPU),
735 REGISTER_KERNEL_BUILDER(Name("MapSize").Device(DEVICE_GPU).HostMemory("size"),
738 Name("OrderedMapSize").Device(DEVICE_GPU).HostMemory("size"),
[all …]
Dstrided_slice_op.cc441 .Device(DEVICE_GPU) \
448 .Device(DEVICE_GPU) \
456 .Device(DEVICE_GPU) \
463 .Device(DEVICE_GPU) \
471 .Device(DEVICE_GPU) \
486 .Device(DEVICE_GPU)
495 .Device(DEVICE_GPU)
505 .Device(DEVICE_GPU)
513 .Device(DEVICE_GPU)
521 .Device(DEVICE_GPU)
/external/tensorflow/tensorflow/core/framework/
Dkernel_def_builder_test.cc37 .Device(DEVICE_GPU) in TEST()
50 .Device(DEVICE_GPU) in TEST()
78 .Device(DEVICE_GPU) in TEST()
95 .Device(DEVICE_GPU) in TEST()
119 .Device(DEVICE_GPU) in TEST()
136 .Device(DEVICE_GPU) in TEST()
160 .Device(DEVICE_GPU) in TEST()
Dvariant_op_registry_test.cc96 DEVICE_GPU, VariantValue,
102 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
224 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, TypeIndex::Make<Blah>()), in TEST()
258 registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, in TEST()
261 DEVICE_GPU, kTypeIndex, f), in TEST()
295 ADD_VARIANT_BINARY_OP, DEVICE_GPU, TypeIndex::Make<Blah>()), in TEST()
330 registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_GPU, kTypeIndex, f); in TEST()
331 EXPECT_DEATH(registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_GPU, in TEST()
Dop_kernel_test.cc73 .Device(::tensorflow::DEVICE_GPU)
102 Name("Test3").Device(DEVICE_GPU).TypeConstraint<float>("T"), TestOp3Cpu);
107 REGISTER_KERNEL_BUILDER(Name("Test4").Device(DEVICE_GPU), DummyKernel);
129 REGISTER_KERNEL_BUILDER(Name("Test5").Device(DEVICE_GPU).Priority(1),
133 return {DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}; in DeviceTypes()
206 ExpectSuccess("Test2", DEVICE_GPU, {DT_INT32}, {DT_INT32}); in TEST_F()
212 ExpectSuccess("Test3", DEVICE_GPU, {DT_FLOAT, DT_FLOAT}, {}); in TEST_F()
237 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
272 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
288 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types_test.cc35 TF_EXPECT_OK(ValidateMemoryTypes(DEVICE_GPU, g)); in TEST()
50 EXPECT_TRUE(errors::IsInternal(ValidateMemoryTypes(DEVICE_GPU, g))); in TEST()
53 TF_EXPECT_OK(EnsureMemoryTypes(DEVICE_GPU, "/device:GPU:0", g)); in TEST()
54 TF_EXPECT_OK(ValidateMemoryTypes(DEVICE_GPU, g)); in TEST()
73 TF_EXPECT_OK(MemoryTypeForOutput(DEVICE_GPU, g, si, 0, &memory_type)); in TEST()

1234567891011