Searched refs:kernel_p (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tensorflow_abi_knowledge_propagation.cc | 92 int kernel_p = 0; in runOnFunction() local
100 kernel_p++; in runOnFunction()
106 Value alloc_ptr = kernel.getArgument(kernel_p); in runOnFunction()
107 Value align_ptr = kernel.getArgument(kernel_p + 1); in runOnFunction()
110 kernel_p + 1, LLVM::LLVMDialect::getAlignAttrName(), in runOnFunction()
115 Value offset = kernel.getArgument(kernel_p + 2); in runOnFunction()
127 kernel.getArgument(kernel_p + 2 + memref.getRank() * 2); in runOnFunction()
142 kernel.setArgAttr(kernel_p + 1, in runOnFunction()
147 kernel_p += memref.getRank() * 2 + 3; in runOnFunction()
|
| D | same_shape_propagation.cc | 331 int kernel_p = 0; in runOnFunction() local
336 kernel_p++; in runOnFunction()
363 .drop_front(kernel_p + 3) in runOnFunction()
372 seen_memrefs.push_back({operand, kernel_p}); in runOnFunction()
374 kernel_p += memref.getRank() * 2 + 3; in runOnFunction()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d_gpu.cu.cc | 34 const int plane, const int height, const int width, const int kernel_p, in MaxPoolGradBackwardNoMaskNCDHW() argument
49 const int pend = min(pstart + kernel_p, plane); in MaxPoolGradBackwardNoMaskNCDHW()
85 const int plane, const int height, const int width, const int kernel_p, in MaxPoolGradBackwardNoMaskNDHWC() argument
104 int pend = min(pstart + kernel_p, plane); in MaxPoolGradBackwardNoMaskNDHWC()
141 const int height, const int width, const int kernel_p, const int kernel_h, in operator ()() argument
153 height, width, kernel_p, kernel_h, kernel_w, stride_p, stride_h, in operator ()()
160 height, width, kernel_p, kernel_h, kernel_w, stride_p, stride_h, in operator ()()
|
| D | pooling_ops_3d_gpu.h | 38 const int width, const int kernel_p, const int kernel_h,
|