| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_neg.cc | 33 REGISTER6(UnaryOp, GPU, "Neg", functor::neg, float, Eigen::half, double, int64,
36 // A special GPU kernel for int32.
|
| cwise_op_sign.cc | 22 REGISTER6(UnaryOp, GPU, "Sign", functor::sign, float, Eigen::half, double,
25 // A special GPU kernel for int32.
|
| cwise_op_square.cc | 23 REGISTER4(UnaryOp, GPU, "Square", functor::square, float, Eigen::half, double,
26 // A special GPU kernel for int32.
|
| cwise_op_squared_difference.cc | 22 REGISTER4(BinaryOp, GPU, "SquaredDifference", functor::squared_difference,
26 // A special GPU kernel for int32.
|
| cwise_op_add_1.cc | 25 REGISTER3(BinaryOp, GPU, "Add", functor::add, float, Eigen::half, double);
26 REGISTER3(BinaryOp, GPU, "AddV2", functor::add, float, Eigen::half, double);
28 // A special GPU kernel for int32.
|
| cwise_op_bitwise_and.cc | 40 REGISTER8(BinaryOp, GPU, "BitwiseAnd", functor::bitwise_and, int8, int16, int32,
|
| cwise_op_bitwise_or.cc | 40 REGISTER8(BinaryOp, GPU, "BitwiseOr", functor::bitwise_or, int8, int16, int32,
|
| cwise_op_bitwise_xor.cc | 40 REGISTER8(BinaryOp, GPU, "BitwiseXor", functor::bitwise_xor, int8, int16, int32,
|
| cwise_op_greater.cc | 22 REGISTER7(BinaryOp, GPU, "Greater", functor::greater, float, Eigen::half,
25 // A special GPU kernel for int32.
|
| cwise_op_greater_equal.cc | 22 REGISTER7(BinaryOp, GPU, "GreaterEqual", functor::greater_equal, float,
25 // A special GPU kernel for int32.
|
| cwise_op_left_shift.cc | 40 REGISTER8(BinaryOp, GPU, "LeftShift", functor::left_shift, int8, int16, int32,
|
| cwise_op_less.cc | 22 REGISTER7(BinaryOp, GPU, "Less", functor::less, float, Eigen::half, double,
25 // A special GPU kernel for int32.
|
| cwise_op_less_equal.cc | 22 REGISTER7(BinaryOp, GPU, "LessEqual", functor::less_equal, float, Eigen::half,
25 // A special GPU kernel for int32.
|
| cwise_op_mul_1.cc | 30 REGISTER4(BinaryOp, GPU, "Mul", functor::mul, float, Eigen::half, double,
32 // A special GPU kernel for int32.
|
| cwise_op_right_shift.cc | 40 REGISTER8(BinaryOp, GPU, "RightShift", functor::right_shift, int8, int16, int32,
|
| cwise_op_sub.cc | 32 REGISTER6(BinaryOp, GPU, "Sub", functor::sub, float, Eigen::half, double, int64,
35 // A special GPU kernel for int32.
|
| sparse_xent_op.cc | 130 REGISTER(GPU, float, int32)
131 REGISTER(GPU, float, int64)
132 REGISTER(GPU, Eigen::half, int32)
133 REGISTER(GPU, Eigen::half, int64)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| tag_constants.py | 36 # Tag for the `gpu` graph.
37 GPU = "gpu"
38 tf_export("saved_model.tag_constants.GPU").export_constant(__name__, "GPU")
47 "GPU",
|
| /external/tensorflow/tensorflow/python/framework/ |
| device.py | 36 # Place the operations on device "GPU:0" in the "ps" job.
37 device_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0)
39 # Both my_var and squared_var will be placed on /job:ps/device:GPU:0.
51 with tf.device(DeviceSpec(job="ps", device_type="GPU", device_index=0):
52 # Nodes created here will be assigned to /job:ps/device:GPU:0.
53 with tf.device(DeviceSpec(device_type="GPU", device_index=1):
54 # Nodes created here will be assigned to /job:train/device:GPU:1.
63 * Device type: The device type string (e.g. "CPU" or "GPU").
75 device_type: Optional device type string (e.g. "CPU" or "GPU")
82 if device_type == "cpu" or device_type == "gpu"
[all...] |
| /hardware/interfaces/thermal/1.0/ |
| types.hal | 24 GPU = 1,
|
| /external/tensorflow/tensorflow/python/eager/ |
| benchmarks_test.py | 20 To run GPU benchmarks:
50 GPU = "/device:GPU:0"
87 # call func to maybe warm up the GPU
105 if device == GPU:
106 # Warmup the GPU
131 self._benchmark_create_tensor([[3.0]], dtypes.float32.as_datatype_enum, GPU)
138 GPU)
141 # int32's are kept on host memory even when executing on GPU.
144 self._benchmark_create_tensor([[3]], dtypes.int32.as_datatype_enum, GPU)
[all...] |
| /external/llvm/lib/Target/AMDGPU/ |
| AMDGPUTargetMachine.cpp | 125 static StringRef getGPUOrDefault(const Triple &TT, StringRef GPU) {
126 if (!GPU.empty())
127 return GPU;
129 // HSA only supports CI+, so change the default GPU to a CI for HSA.
185 StringRef GPU = getGPUName(F);
188 SmallString<128> SubtargetKey(GPU);
197 I = llvm::make_unique<R600Subtarget>(TargetTriple, GPU, FS, *this);
226 StringRef GPU = getGPUName(F);
229 SmallString<128> SubtargetKey(GPU);
238 I = llvm::make_unique<SISubtarget>(TargetTriple, GPU, FS, *this)
[all...] |
| /external/skia/src/gpu/vk/ |
| GrVkStencilAttachment.cpp | 14 #define VK_CALL(GPU, X) GR_VK_CALL(GPU->vkInterface(), X)
16 GrVkStencilAttachment::GrVkStencilAttachment(GrVkGpu* gpu,
21 : GrStencilAttachment(gpu, desc.fWidth, desc.fHeight, format.fStencilBits, desc.fSamples)
29 GrVkStencilAttachment* GrVkStencilAttachment::Create(GrVkGpu* gpu,
47 if (!GrVkImage::InitImageInfo(gpu, imageDesc, &info)) {
51 const GrVkImageView* imageView = GrVkImageView::Create(gpu, info.fImage,
55 GrVkImage::DestroyImageInfo(gpu, &info);
59 GrVkStencilAttachment* stencil = new GrVkStencilAttachment(gpu, format, imageDesc,
61 imageView->unref(gpu);
80 GrVkGpu* gpu = this->getVkGpu(); local
[all...] |
| GrVkUniformBuffer.cpp | 11 #define VK_CALL(GPU, X) GR_VK_CALL(GPU->vkInterface(), X)
13 GrVkUniformBuffer* GrVkUniformBuffer::Create(GrVkGpu* gpu, size_t size) {
19 resource = gpu->resourceProvider().findOrCreateStandardUniformBufferResource();
21 resource = CreateResource(gpu, size);
31 GrVkUniformBuffer* buffer = new GrVkUniformBuffer(gpu, desc,
36 resource->unref(gpu);
42 const GrVkResource* GrVkUniformBuffer::CreateResource(GrVkGpu* gpu, size_t size) {
62 err = VK_CALL(gpu, CreateBuffer(gpu->device(), &bufInfo, nullptr, &buffer))
[all...] |
| /external/skqp/src/gpu/vk/ |
| GrVkStencilAttachment.cpp | 14 #define VK_CALL(GPU, X) GR_VK_CALL(GPU->vkInterface(), X)
16 GrVkStencilAttachment::GrVkStencilAttachment(GrVkGpu* gpu,
21 : GrStencilAttachment(gpu, desc.fWidth, desc.fHeight, format.fStencilBits, desc.fSamples)
29 GrVkStencilAttachment* GrVkStencilAttachment::Create(GrVkGpu* gpu,
47 if (!GrVkImage::InitImageInfo(gpu, imageDesc, &info)) {
51 const GrVkImageView* imageView = GrVkImageView::Create(gpu, info.fImage,
55 GrVkImage::DestroyImageInfo(gpu, &info);
59 GrVkStencilAttachment* stencil = new GrVkStencilAttachment(gpu, format, imageDesc,
61 imageView->unref(gpu);
80 GrVkGpu* gpu = this->getVkGpu(); local
[all...] |
