Searched defs:dilation_w (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 58 dilation_w, dilation_h, padding, act): argument
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 588 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvBackpropCommonOp() local
|
| D | mkl_conv_ops.cc | 470 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvOp() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 288 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in Conv2DBackpropFilterOp() local
|
| D | conv_ops.cc | 386 const int64 dilation_w = GetTensorDim(dilations, data_format, 'W'); in InitConv2DParameters() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 887 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
978 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| D | legalize_tfl.cc | 1137 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
1353 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
|
| D | legalize_common.cc | 2831 int64_t dilation_w = dilations_attr[2].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 85 absl::Status CheckDilation(int dilation_h, int dilation_w) { in CheckDilation()
95 int dilation_h, int dilation_w) { in CheckStridesAndDilation()
|