Home
last modified time | relevance | path

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

12345678

/external/tensorflow/tensorflow/core/kernels/
Dshape_ops.cc73 .Device(DEVICE_GPU) \
79 .Device(DEVICE_GPU) \
94 .Device(DEVICE_GPU)
101 .Device(DEVICE_GPU)
125 .Device(DEVICE_GPU) \
131 .Device(DEVICE_GPU) \
145 .Device(DEVICE_GPU)
152 .Device(DEVICE_GPU)
227 .Device(DEVICE_GPU) \
239 .Device(DEVICE_GPU)
[all …]
Dcontrol_flow_ops.cc58 .Device(DEVICE_GPU) \
65 .Device(DEVICE_GPU) \
87 .Device(DEVICE_GPU) \
97 .Device(DEVICE_GPU) \
244 .Device(DEVICE_GPU) \
251 .Device(DEVICE_GPU) \
292 .Device(DEVICE_GPU) \
299 .Device(DEVICE_GPU) \
349 Name("Enter").Device(DEVICE_GPU).TypeConstraint<type>("T"), EnterOp)
352 Name("RefEnter").Device(DEVICE_GPU).TypeConstraint<type>("T"), EnterOp)
[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) \
117 .Device(DEVICE_GPU) \
123 .Device(DEVICE_GPU) \
Dfft_ops.cc469 REGISTER_KERNEL_BUILDER(Name("FFT").Device(DEVICE_GPU), FFTGPU<true, false, 1>);
470 REGISTER_KERNEL_BUILDER(Name("IFFT").Device(DEVICE_GPU),
472 REGISTER_KERNEL_BUILDER(Name("FFT2D").Device(DEVICE_GPU),
474 REGISTER_KERNEL_BUILDER(Name("IFFT2D").Device(DEVICE_GPU),
476 REGISTER_KERNEL_BUILDER(Name("FFT3D").Device(DEVICE_GPU),
478 REGISTER_KERNEL_BUILDER(Name("IFFT3D").Device(DEVICE_GPU),
482 Name("RFFT").Device(DEVICE_GPU).HostMemory("fft_length"),
485 Name("IRFFT").Device(DEVICE_GPU).HostMemory("fft_length"),
488 Name("RFFT2D").Device(DEVICE_GPU).HostMemory("fft_length"),
491 Name("IRFFT2D").Device(DEVICE_GPU).HostMemory("fft_length"),
[all …]
Dtensor_array_ops.cc240 .Device(DEVICE_GPU) \
246 .Device(DEVICE_GPU) \
252 .Device(DEVICE_GPU) \
366 .Device(DEVICE_GPU)
371 .Device(DEVICE_GPU)
376 .Device(DEVICE_GPU)
438 .Device(DEVICE_GPU) \
444 .Device(DEVICE_GPU) \
450 .Device(DEVICE_GPU) \
527 .Device(DEVICE_GPU) \
[all …]
Drelu_op.cc133 Name("Relu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
136 Name("ReluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
139 Name("Relu6").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
142 Name("Relu6Grad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
145 Name("Elu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
148 Name("EluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
151 Name("Selu").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
154 Name("SeluGrad").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
Dfunction_ops.cc121 Name("_Arg").Device(DEVICE_GPU).TypeConstraint<type>("T"), ArgOp);
124 .Device(DEVICE_GPU)
131 .Device(DEVICE_GPU)
138 Name("_Retval").Device(DEVICE_GPU).TypeConstraint<type>("T"), RetvalOp);
141 .Device(DEVICE_GPU)
174 Name("_ListToArray").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
177 Name("_ArrayToList").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
187 .Device(DEVICE_GPU)
193 .Device(DEVICE_GPU)
279 REGISTER_KERNEL_BUILDER(Name(kGradientOp).Device(DEVICE_GPU),
[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) \
Dlist_kernels.cu.cc42 .Device(DEVICE_GPU), \
57 .Device(DEVICE_GPU) \
70 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
74 DEVICE_GPU, TensorList,
Dstack_ops.cc229 REGISTER_KERNEL_BUILDER(Name("Stack").Device(DEVICE_GPU).HostMemory("handle"),
233 .Device(DEVICE_GPU)
334 .Device(DEVICE_GPU) \
339 .Device(DEVICE_GPU) \
352 .Device(DEVICE_GPU) \
359 .Device(DEVICE_GPU) \
443 .Device(DEVICE_GPU) \
448 .Device(DEVICE_GPU) \
461 .Device(DEVICE_GPU) \
467 .Device(DEVICE_GPU) \
[all …]
Dreshape_op.cc34 .Device(DEVICE_GPU) \
40 .Device(DEVICE_GPU) \
94 .Device(DEVICE_GPU)
102 .Device(DEVICE_GPU)
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) \
62 .Device(DEVICE_GPU)
71 .Device(DEVICE_GPU)
Dpad_op.cc223 .Device(DEVICE_GPU) \
229 .Device(DEVICE_GPU) \
235 .Device(DEVICE_GPU) \
242 .Device(DEVICE_GPU) \
255 .Device(DEVICE_GPU)
263 .Device(DEVICE_GPU)
271 .Device(DEVICE_GPU)
280 .Device(DEVICE_GPU)
Dsoftmax_op_gpu.cu.cc150 Name("Softmax").Device(DEVICE_GPU).TypeConstraint<Eigen::half>("T"),
153 Name("Softmax").Device(DEVICE_GPU).TypeConstraint<float>("T"),
156 Name("Softmax").Device(DEVICE_GPU).TypeConstraint<double>("T"),
159 Name("LogSoftmax").Device(DEVICE_GPU).TypeConstraint<Eigen::half>("T"),
162 Name("LogSoftmax").Device(DEVICE_GPU).TypeConstraint<float>("T"),
Dreduction_ops_sum.cc41 .Device(DEVICE_GPU) \
48 .Device(DEVICE_GPU) \
63 .Device(DEVICE_GPU)
72 .Device(DEVICE_GPU)
Dreduction_ops_min.cc41 .Device(DEVICE_GPU) \
48 .Device(DEVICE_GPU) \
62 .Device(DEVICE_GPU)
71 .Device(DEVICE_GPU)
Dmap_stage_op.cc546 Name("MapStage").HostMemory("key").HostMemory("indices").Device(DEVICE_GPU),
551 .Device(DEVICE_GPU),
609 .Device(DEVICE_GPU),
614 .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 …]
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)
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/contrib/nccl/kernels/
Dnccl_ops.cc111 REGISTER_KERNEL_BUILDER(Name("NcclAllReduce").Device(DEVICE_GPU),
136 REGISTER_KERNEL_BUILDER(Name("_NcclReduceSend").Device(DEVICE_GPU),
168 REGISTER_KERNEL_BUILDER(Name("_NcclReduceRecv").Device(DEVICE_GPU),
193 REGISTER_KERNEL_BUILDER(Name("_NcclBroadcastSend").Device(DEVICE_GPU),
226 Name("_NcclBroadcastRecv").Device(DEVICE_GPU).HostMemory("shape"),
239 REGISTER_KERNEL_BUILDER(Name("NcclBroadcast").Device(DEVICE_GPU),
241 REGISTER_KERNEL_BUILDER(Name("NcclReduce").Device(DEVICE_GPU), NcclStubKernel);
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry_test.cc106 DEVICE_GPU, VariantValue,
114 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
245 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, "YOU SHALL NOT PASS"), in TEST()
279 registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, kTypeName, in TEST()
282 DEVICE_GPU, kTypeName, f), in TEST()
315 ADD_VARIANT_BINARY_OP, DEVICE_GPU, "YOU SHALL NOT PASS"), in TEST()
349 registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_GPU, kTypeName, f); in TEST()
350 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);
106 return {DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}; in DeviceTypes()
177 ExpectSuccess("Test2", DEVICE_GPU, {DT_INT32}, {DT_INT32}); in TEST_F()
183 ExpectSuccess("Test3", DEVICE_GPU, {DT_FLOAT, DT_FLOAT}, {}); in TEST_F()
211 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0]); in TEST_F()
227 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0]); in TEST_F()
237 DEVICE_GPU, not_found); in TEST_F()
239 DEVICE_GPU, not_found); in TEST_F()
[all …]

12345678