Home
last modified time | relevance | path

Searched refs:DepthwiseArgs (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h25 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,
Ddepthwise_conv_op_gpu.h45 const DepthwiseArgs& args) { in CanLaunchDepthwiseConv2dGPUSmall()
58 const DepthwiseArgs& args, const int block_height) { in CanLaunchDepthwiseConv2dBackpropFilterGPUSmall()
76 DepthwiseConv2dGPUKernelNHWC(const DepthwiseArgs args, const T* input, in DepthwiseConv2dGPUKernelNHWC()
177 const DepthwiseArgs args, const T* input, const T* filter, T* output) { in DepthwiseConv2dGPUKernelNHWCSmall()
316 DepthwiseConv2dGPUKernelNCHW(const DepthwiseArgs args, const T* input, in DepthwiseConv2dGPUKernelNCHW()
461 const DepthwiseArgs args, const T* input, const T* filter, T* output) { in DepthwiseConv2dGPUKernelNCHWSmall()
601 const DepthwiseArgs& args, const T* input, in LaunchDepthwiseConv2dGPUSmall()
607 void (*kernel)(const DepthwiseArgs, const T*, const T*, T*); in LaunchDepthwiseConv2dGPUSmall()
679 const DepthwiseArgs& args, const T* input,
699 const DepthwiseArgs& args, const T* input,
[all …]
Ddepthwise_conv_grad_op.cc132 DepthwiseArgs args; \
180 static void CopyOutputBackpropRegion(const DepthwiseArgs& args, in CopyOutputBackpropRegion()
276 static void ComputeBackpropInput(const DepthwiseArgs& args, in ComputeBackpropInput()
374 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args, in operator ()()
460 static void DepthwiseConvBackpropInputReference(const DepthwiseArgs& args, in DepthwiseConvBackpropInputReference()
746 static void ComputeBackpropFilter(const DepthwiseArgs& args, in ComputeBackpropFilter()
828 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args, in operator ()()
942 static void DepthwiseConvBackpropFilterReference(const DepthwiseArgs& args, in DepthwiseConvBackpropFilterReference()
Ddepthwise_conv_op.cc82 static void Run(const DepthwiseArgs& args, in Run()
157 void operator()(OpKernelContext* ctx, const DepthwiseArgs& args, in operator ()()
412 DepthwiseArgs args; in Compute()