Searched refs:TypedKernel (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| D | asm_compiler.h | 95 port::StatusOr<std::shared_ptr<TypedKernel<Args...>>> LoadKernelOrGetPtr( in LoadKernelOrGetPtr()
104 std::shared_ptr<TypedKernel<Args...>>>(); in LoadKernelOrGetPtr()
112 std::shared_ptr<TypedKernel<Args...>> loaded, in LoadKernelOrGetPtr()
|
| D | redzone_allocator.cc | 178 TypedKernel<DeviceMemory<uint8>, uint8, uint64_t, DeviceMemory<uint64_t>>;
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | kernel.h | 510 class TypedKernel : public KernelBase {
515 explicit TypedKernel(StreamExecutor *parent) : KernelBase(parent) {}
519 TypedKernel(StreamExecutor *parent, internal::KernelInterface *implementation)
600 SE_DISALLOW_COPY_AND_ASSIGN(TypedKernel);
720 struct KernelParamsOk<TypedKernel<Params...>, Args...> {
|
| D | stream_executor_pimpl.h | 461 port::StatusOr<std::unique_ptr<TypedKernel<Args...>>> CreateTypedKernel(
557 friend class TypedKernel; variable
810 inline port::StatusOr<std::unique_ptr<TypedKernel<Args...>>>
814 auto kernel_base = std::make_unique<TypedKernel<Args...>>(this); in CreateTypedKernel()
|
| D | stream.h | 192 const TypedKernel<Params...> &kernel, Args... args);
1625 const TypedKernel<Params...> &kernel, in ThenLaunch()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | precompiled_kernels.cc | 106 StatusOr<se::TypedKernel<KernelArgs...>*> Get( in Get()
152 std::unique_ptr<se::TypedKernel<KernelArgs...>>>
|
| D | buffer_comparator.cc | 603 se::TypedKernel<se::DeviceMemory<ElementT>, se::DeviceMemory<ElementT>,
|