Searched defs:dilation_h (Results 1 – 10 of 10) 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/lite/tools/versioning/ |
| D | gpu_compatibility.cc | 189 absl::Status CheckDilation(int dilation_h, int dilation_w) { in CheckDilation()
199 int dilation_h, int dilation_w) { in CheckStridesAndDilation()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 614 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvBackpropCommonOp() local
|
| D | mkl_conv_ops.cc | 582 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvOp() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 290 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DBackpropFilterOp() local
|
| D | conv_ops.cc | 519 const int64_t dilation_h = GetTensorDim(dilations, data_format, 'H'); in InitConv2DParameters() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 888 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
982 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| D | legalize_tfl.cc | 969 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
1201 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
|
| D | legalize_common.cc | 3082 int64_t dilation_h = dilations_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1093 auto dilation_h = op.dilation_h_factor().getInt(); in inferReturnTypes() local
|