Home
last modified time | relevance | path

Searched defs:dilation_h (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dops_defs.py58 dilation_w, dilation_h, padding, act): argument
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h587 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvBackpropCommonOp() local
Dmkl_conv_ops.cc469 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvOp() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_filter_ops.cc287 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DBackpropFilterOp() local
Dconv_ops.cc385 const int64 dilation_h = GetTensorDim(dilations, data_format, 'H'); in InitConv2DParameters() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc886 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
977 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
Dlegalize_tfl.cc1136 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
1352 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
Dlegalize_common.cc2830 int64_t dilation_h = dilations_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc85 absl::Status CheckDilation(int dilation_h, int dilation_w) { in CheckDilation()
95 int dilation_h, int dilation_w) { in CheckStridesAndDilation()