Searched refs:DepthwiseArgs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.h | 25 struct DepthwiseArgs { struct
43 DepthwiseArgs() in DepthwiseArgs() argument
64 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args,
71 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args,
78 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args,
86 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args,
93 void operator()(class OpKernelContext* ctx, const DepthwiseArgs& args,
100 void operator()(class OpKernelContext* ctx, const DepthwiseArgs& args,
131 void operator()(const DepthwiseArgs& args, const T* filter,
192 void operator()(const DepthwiseArgs& args,
|
| D | depthwise_conv_op_gpu.h | 55 const DepthwiseArgs& args) {
68 const DepthwiseArgs& args, const int block_height) {
86 DepthwiseConv2dGPUKernelNHWC(const DepthwiseArgs args,
192 const DepthwiseArgs args, const T* __restrict__ input,
333 DepthwiseConv2dGPUKernelNCHW(const DepthwiseArgs args,
483 const DepthwiseArgs args, const T* __restrict__ input,
625 const DepthwiseArgs& args, const T* input,
632 void (*kernel)(const DepthwiseArgs, const T*, const T*, T*);
689 const DepthwiseArgs& args, const T* input,
707 const DepthwiseArgs& args, const T* input,
[all …]
|
| D | depthwise_conv_grad_op.cc | 144 DepthwiseArgs args; \
192 static void CopyOutputBackpropRegion(const DepthwiseArgs& args, in CopyOutputBackpropRegion()
288 static void ComputeBackpropInput(const DepthwiseArgs& args, in ComputeBackpropInput()
386 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args, in operator ()()
472 static void DepthwiseConvBackpropInputReference(const DepthwiseArgs& args, in DepthwiseConvBackpropInputReference()
787 static void ComputeBackpropFilter(const DepthwiseArgs& args, in ComputeBackpropFilter()
869 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args, in operator ()()
983 static void DepthwiseConvBackpropFilterReference(const DepthwiseArgs& args, in DepthwiseConvBackpropFilterReference()
|
| D | depthwise_conv_op.cc | 87 static void Run(const DepthwiseArgs& args, in Run()
162 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args, in operator ()()
451 DepthwiseArgs args; in Compute()
|