Searched defs:dilation_w (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
| D | tf_to_xla_attribute_utils.cc | 97 const int dilation_w = dilations[2].cast<IntegerAttr>().getInt(); in PadForDynamicShapedInputSamePadding() local
173 const int dilation_w = dilations[2].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 | 65 const int dilation_w = dilations[2].cast<IntegerAttr>().getInt(); in PrepareXlaConvParams() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 643 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvBackpropCommonOp() local
|
| D | mkl_conv_ops.cc | 598 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvOp() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 292 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in Conv2DBackpropFilterOp() local
|
| D | conv_ops.cc | 531 const int64_t dilation_w = GetTensorDim(dilations, data_format, 'W'); in InitConv2DParameters() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 890 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
981 int64_t dilation_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| D | legalize_tfl.cc | 1038 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
1270 int64_t dilation_w = tfl_conv2d_op.dilation_w_factor(); in matchAndRewrite() local
|
| D | legalize_common.cc | 3361 int64_t dilation_w = dilations_attr[2].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1236 auto dilation_w = op.dilation_w_factorAttr().getInt(); in inferReturnTypes() local
|