Searched refs:ScanOp (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scan_ops.cc | 37 class ScanOp : public OpKernel { class
39 explicit ScanOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in ScanOp() function in tensorflow::ScanOp
127 ScanOp<CPUDevice, type, Eigen::internal::SumReducer<type>, int32>) \
133 ScanOp<CPUDevice, type, Eigen::internal::SumReducer<type>, int64>)
145 ScanOp<GPUDevice, type, Eigen::internal::SumReducer<type>, int32>) \
152 ScanOp<GPUDevice, type, Eigen::internal::SumReducer<type>, int64>)
166 ScanOp<CPUDevice, type, Eigen::internal::ProdReducer<type>, int32>) \
172 ScanOp<CPUDevice, type, Eigen::internal::ProdReducer<type>, int64>)
184 ScanOp<GPUDevice, type, Eigen::internal::ProdReducer<type>, int32>) \
191 ScanOp<GPUDevice, type, Eigen::internal::ProdReducer<type>, int64>)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | scan_ops.cc | 42 class ScanOp : public XlaOpKernel { class
44 ScanOp(OpKernelConstruction* ctx, bool sum) : XlaOpKernel(ctx), sum_(sum) { in ScanOp() function in tensorflow::__anon162cdad30111::ScanOp
129 class CumsumOp : public ScanOp {
131 explicit CumsumOp(OpKernelConstruction* ctx) : ScanOp(ctx, /*sum=*/true) {} in CumsumOp()
138 class CumprodOp : public ScanOp {
140 explicit CumprodOp(OpKernelConstruction* ctx) : ScanOp(ctx, /*sum=*/false) {} in CumprodOp()
|
| /external/llvm-project/llvm/lib/Target/AMDGPU/ |
| D | AMDGPUAtomicOptimizer.cpp | 489 const AtomicRMWInst::BinOp ScanOp = in optimizeAtomic() local
491 NewV = buildScan(B, ScanOp, NewV, Identity); in optimizeAtomic()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/ |
| D | AMDGPUAtomicOptimizer.cpp | 485 const AtomicRMWInst::BinOp ScanOp = in optimizeAtomic() local
487 NewV = buildScan(B, ScanOp, NewV, Identity); in optimizeAtomic()
|