Searched refs:convFwdDims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.cc | 104 explicit MklConvFwdPrimitive(const MklConvFwdParams& convFwdDims) in MklConvFwdPrimitive() argument
109 Setup(convFwdDims); in MklConvFwdPrimitive()
214 void Setup(const MklConvFwdParams& convFwdDims) { in Setup() argument
217 {convFwdDims.src_dims}, MklDnnType<Tinput>(), memory::format::any)); in Setup()
220 {convFwdDims.filter_dims}, MklDnnType<Tfilter>(), memory::format::any)); in Setup()
223 {convFwdDims.dst_dims}, MklDnnType<Toutput>(), memory::format::any)); in Setup()
225 if (!convFwdDims.bias_dims.empty()) in Setup()
227 {convFwdDims.bias_dims}, MklDnnType<Tbias>(), memory::format::any)); in Setup()
230 if (!convFwdDims.bias_dims.empty()) { in Setup()
234 convFwdDims.strides, convFwdDims.dilations, convFwdDims.padding_left, in Setup()
[all …]
|