Home
last modified time | relevance | path

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

123456789

/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.cc74 .Device(DEVICE_GPU) \
80 .Device(DEVICE_GPU) \
95 .Device(DEVICE_GPU)
102 .Device(DEVICE_GPU)
126 .Device(DEVICE_GPU) \
132 .Device(DEVICE_GPU) \
146 .Device(DEVICE_GPU)
153 .Device(DEVICE_GPU)
228 .Device(DEVICE_GPU) \
240 .Device(DEVICE_GPU)
[all …]
Dstack_ops.cc44 REGISTER_KERNEL_BUILDER(Name("Stack").Device(DEVICE_GPU).HostMemory("handle"),
48 .Device(DEVICE_GPU)
69 .Device(DEVICE_GPU) \
74 .Device(DEVICE_GPU) \
87 .Device(DEVICE_GPU) \
94 .Device(DEVICE_GPU) \
136 .Device(DEVICE_GPU) \
141 .Device(DEVICE_GPU) \
154 .Device(DEVICE_GPU) \
160 .Device(DEVICE_GPU) \
[all …]
Didentity_op.cc90 Name("Identity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
93 Name("PreventGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
96 Name("RefIdentity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
99 Name("StopGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
102 .Device(DEVICE_GPU) \
106 .Device(DEVICE_GPU) \
121 .Device(DEVICE_GPU) \
127 .Device(DEVICE_GPU) \
133 .Device(DEVICE_GPU) \
139 .Device(DEVICE_GPU) \
Dcontrol_flow_ops.cc58 .Device(DEVICE_GPU) \
65 .Device(DEVICE_GPU) \
93 .Device(DEVICE_GPU) \
103 .Device(DEVICE_GPU) \
251 .Device(DEVICE_GPU) \
258 .Device(DEVICE_GPU) \
303 .Device(DEVICE_GPU) \
310 .Device(DEVICE_GPU) \
360 Name("Enter").Device(DEVICE_GPU).TypeConstraint<type>("T"), EnterOp)
363 Name("RefEnter").Device(DEVICE_GPU).TypeConstraint<type>("T"), EnterOp)
[all …]
Dfunction_ops.cc101 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), ArgOp);
105 .Device(DEVICE_GPU)
110 Name(kDeviceArgOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), ArgOp);
114 .Device(DEVICE_GPU)
120 .Device(DEVICE_GPU)
126 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<Variant>("T"), ArgOp);
130 Name(kRetOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), RetvalOp);
135 .Device(DEVICE_GPU)
140 Name(kDeviceRetOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), RetvalOp);
143 .Device(DEVICE_GPU)
[all …]
Dtensor_array_ops.cc241 .Device(DEVICE_GPU) \
247 .Device(DEVICE_GPU) \
253 .Device(DEVICE_GPU) \
392 .Device(DEVICE_GPU)
397 .Device(DEVICE_GPU)
402 .Device(DEVICE_GPU)
407 .Device(DEVICE_GPU)
470 .Device(DEVICE_GPU) \
476 .Device(DEVICE_GPU) \
482 .Device(DEVICE_GPU) \
[all …]
Drelu_op.cc158 Name("Relu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
161 Name("ReluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
164 Name("Relu6").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
167 Name("Relu6Grad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
170 Name("LeakyRelu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
173 Name("LeakyReluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
176 Name("Elu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
179 Name("EluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
182 Name("Selu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
185 Name("SeluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
[all …]
Dfft_ops.cc506 REGISTER_KERNEL_BUILDER(Name("FFT").Device(DEVICE_GPU), FFTGPU<true, false, 1>);
507 REGISTER_KERNEL_BUILDER(Name("IFFT").Device(DEVICE_GPU),
509 REGISTER_KERNEL_BUILDER(Name("FFT2D").Device(DEVICE_GPU),
511 REGISTER_KERNEL_BUILDER(Name("IFFT2D").Device(DEVICE_GPU),
513 REGISTER_KERNEL_BUILDER(Name("FFT3D").Device(DEVICE_GPU),
515 REGISTER_KERNEL_BUILDER(Name("IFFT3D").Device(DEVICE_GPU),
519 Name("RFFT").Device(DEVICE_GPU).HostMemory("fft_length"),
522 Name("IRFFT").Device(DEVICE_GPU).HostMemory("fft_length"),
525 Name("RFFT2D").Device(DEVICE_GPU).HostMemory("fft_length"),
528 Name("IRFFT2D").Device(DEVICE_GPU).HostMemory("fft_length"),
[all …]
Ddebug_ops.cc32 REGISTER_KERNEL_BUILDER(Name("Copy").Device(DEVICE_GPU), CopyOp);
35 .Device(DEVICE_GPU)
57 .Device(DEVICE_GPU)
81 .Device(DEVICE_GPU) \
117 .Device(DEVICE_GPU) \
Dinplace_ops.cc220 .Device(DEVICE_GPU) \
228 Name("ParallelConcat").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
235 .Device(DEVICE_GPU) \
245 .Device(DEVICE_GPU)
491 Name("InplaceUpdate").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \
494 Name("InplaceAdd").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \
497 Name("InplaceSub").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \
500 Name("DeepCopy").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \
504 Name("InplaceUpdate").Device(DEVICE_GPU).TypeConstraint<bool>("T"),
512 .Device(DEVICE_GPU)
[all …]
Dreshape_op.cc34 .Device(DEVICE_GPU) \
40 .Device(DEVICE_GPU) \
94 .Device(DEVICE_GPU)
102 .Device(DEVICE_GPU)
Dnccl_ops.cc119 REGISTER_KERNEL_BUILDER(Name("NcclAllReduce").Device(DEVICE_GPU),
151 REGISTER_KERNEL_BUILDER(Name("_NcclReduceSend").Device(DEVICE_GPU),
191 REGISTER_KERNEL_BUILDER(Name("_NcclReduceRecv").Device(DEVICE_GPU),
221 REGISTER_KERNEL_BUILDER(Name("_NcclBroadcastSend").Device(DEVICE_GPU),
259 Name("_NcclBroadcastRecv").Device(DEVICE_GPU).HostMemory("shape"),
272 REGISTER_KERNEL_BUILDER(Name("NcclBroadcast").Device(DEVICE_GPU),
274 REGISTER_KERNEL_BUILDER(Name("NcclReduce").Device(DEVICE_GPU), NcclStubKernel);
Dcwise_op_conj.cc27 Name("Conj").Device(DEVICE_GPU).TypeConstraint<Variant>("T"),
30 Name("Conj").Device(DEVICE_GPU).TypeConstraint<complex64>("T"),
33 Name("Conj").Device(DEVICE_GPU).TypeConstraint<complex128>("T"),
Dreduction_ops_max.cc41 .Device(DEVICE_GPU) \
48 .Device(DEVICE_GPU) \
64 .Device(DEVICE_GPU)
73 .Device(DEVICE_GPU)
Dreduction_ops_min.cc41 .Device(DEVICE_GPU) \
48 .Device(DEVICE_GPU) \
62 .Device(DEVICE_GPU)
71 .Device(DEVICE_GPU)
Dreduction_ops_sum.cc41 .Device(DEVICE_GPU) \
48 .Device(DEVICE_GPU) \
64 .Device(DEVICE_GPU)
73 .Device(DEVICE_GPU)
Dcwise_op_equal_to_1.cc31 Name("ApproximateEqual").Device(DEVICE_GPU).TypeConstraint<float>("T"),
34 Name("ApproximateEqual").Device(DEVICE_GPU).TypeConstraint<double>("T"),
41 .Device(DEVICE_GPU)
Dmap_stage_op.cc548 Name("MapStage").HostMemory("key").HostMemory("indices").Device(DEVICE_GPU),
553 .Device(DEVICE_GPU),
610 .Device(DEVICE_GPU),
615 .Device(DEVICE_GPU),
668 Name("MapPeek").HostMemory("key").HostMemory("indices").Device(DEVICE_GPU),
673 .Device(DEVICE_GPU),
733 .Device(DEVICE_GPU),
738 .Device(DEVICE_GPU),
779 REGISTER_KERNEL_BUILDER(Name("MapSize").Device(DEVICE_GPU).HostMemory("size"),
782 Name("OrderedMapSize").Device(DEVICE_GPU).HostMemory("size"),
[all …]
Dops_testutil.cc31 if (device_type == DEVICE_GPU) { in SetDevice()
39 CHECK_NE(device_type, DEVICE_GPU) in SetDevice()
48 if (device_type_ == DEVICE_GPU) { in GetOutput()
Dsession_ops.cc74 .Device(DEVICE_GPU) \
79 .Device(DEVICE_GPU) \
127 .Device(DEVICE_GPU) \
166 Name("DeleteSessionTensor").Device(DEVICE_GPU).HostMemory("handle"),
/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types_test.cc35 TF_EXPECT_OK(ValidateMemoryTypes(DEVICE_GPU, g)); in TEST()
53 EXPECT_TRUE(errors::IsInternal(ValidateMemoryTypes(DEVICE_GPU, g))); in TEST()
56 TF_EXPECT_OK(EnsureMemoryTypes(DEVICE_GPU, "/device:GPU:0", g)); in TEST()
57 TF_EXPECT_OK(ValidateMemoryTypes(DEVICE_GPU, g)); in TEST()
85 TF_EXPECT_OK(MemoryTypeForOutput(DEVICE_GPU, g, si, 0, &memory_type)); in TEST()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry_test.cc96 DEVICE_GPU, VariantValue,
102 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
206 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, MakeTypeIndex<Blah>()), in TEST()
241 registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, in TEST()
244 DEVICE_GPU, kTypeIndex, f), in TEST()
278 ADD_VARIANT_BINARY_OP, DEVICE_GPU, MakeTypeIndex<Blah>()), in TEST()
313 registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_GPU, kTypeIndex, f); in TEST()
314 EXPECT_DEATH(registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_GPU, in TEST()
Dop_kernel_test.cc69 .Device(::tensorflow::DEVICE_GPU)
98 Name("Test3").Device(DEVICE_GPU).TypeConstraint<float>("T"), TestOp3Cpu);
103 REGISTER_KERNEL_BUILDER(Name("Test4").Device(DEVICE_GPU), DummyKernel);
123 REGISTER_KERNEL_BUILDER(Name("Test5").Device(DEVICE_GPU).Priority(1),
127 return {DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}; in DeviceTypes()
198 ExpectSuccess("Test2", DEVICE_GPU, {DT_INT32}, {DT_INT32}); in TEST_F()
204 ExpectSuccess("Test3", DEVICE_GPU, {DT_FLOAT, DT_FLOAT}, {}); in TEST_F()
232 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
248 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
260 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[1].first); in TEST_F()
[all …]
Dkernel_def_builder_test.cc36 .Device(DEVICE_GPU) in TEST()
49 .Device(DEVICE_GPU) in TEST()
77 .Device(DEVICE_GPU) in TEST()

123456789