Searched defs:dilation_h (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
| D | tf_to_xla_attribute_utils.cc | 96 const int dilation_h = dilations[1].cast<IntegerAttr>().getInt(); in PadForDynamicShapedInputSamePadding() local
172 const int dilation_h = dilations[1].cast<IntegerAttr>().getInt(); in CalculatePaddingAndPadIfNeeded() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 54 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/compiler/mlir/quantization/tensorflow/passes/ |
| D | replace_cast_hacks_with_tf_xla_ops.cc | 64 const int dilation_h = dilations[1].cast<IntegerAttr>().getInt(); in PrepareXlaConvParams() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 642 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvBackpropCommonOp() local
|
| D | mkl_conv_ops.cc | 597 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvOp() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 291 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DBackpropFilterOp() local
|
| D | conv_ops.cc | 530 const int64_t dilation_h = GetTensorDim(dilations, data_format, 'H'); in InitConv2DParameters() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 889 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
980 int64_t dilation_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| D | legalize_tfl.cc | 1037 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
1269 int64_t dilation_h = tfl_conv2d_op.dilation_h_factor(); in matchAndRewrite() local
|
| D | legalize_common.cc | 3360 int64_t dilation_h = dilations_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1235 auto dilation_h = op.dilation_h_factorAttr().getInt(); in inferReturnTypes() local
|