| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | depthwise_conv_op_base.py | 70 def _DepthwiseConv2dNumpy(x1, x2, strides, padding, data_format, dilations): argument
151 dilations=None): argument
187 dilations=None): argument
235 dilations=None): argument
269 dilations=None): argument
308 dilations=None, argument
647 dilations=None): argument
|
| D | conv_ops_test.py | 182 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, dilations, argument
304 padding, dilations, rtol=1e-4): argument
328 dilations=(1, 1), argument
378 dilations=(1, 1), argument
791 def _VerifyGroupConvFwd(self, tensor_in_sizes, filter_in_sizes, dilations, argument
921 dilations=(1, 1)): argument
1143 dilations=(1, 1), argument
1317 output_sizes, strides, dilations, argument
1363 output_sizes, strides, dilations, argument
1566 dilations=(1, 1), argument
[all …]
|
| D | cudnn_deterministic_base.py | 77 def _random_out_op(self, in_shape, filter_shape, strides, padding, dilations): argument
|
| D | conv_ops_3d_test.py | 167 padding, dilations): argument
787 strides, dilations, padding, data_format, use_gpu, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 485 dilations, argument
558 dilations=None): argument
1039 dilations=None): # pylint: disable=g-doc-args argument
1179 dilations=None, argument
1203 dilations=None, argument
1513 dilations=None): argument
1692 dilations=None, argument
1997 dilations=None): argument
2101 dilations=None, argument
2172 dilations=None, argument
[all …]
|
| D | nn_impl.py | 776 dilations=None): argument
903 dilations=None, argument
1007 dilations=None): argument
1114 dilations=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | replace_cast_hacks_with_tf_xla_ops.cc | 56 ArrayAttr dilations, int feature_group_cnt, in PrepareXlaConvParams()
98 ArrayAttr strides, ArrayAttr dilations, in CreateXLAConvOp()
164 ArrayAttr dilations, in CreateXLAConvOpFromTFConv2DOp()
185 Value conv_output, ArrayAttr strides, ArrayAttr dilations, in CreateXLAConvOpFromTFDepthwiseConv2DOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 56 const gtl::ArraySlice<int32> dilations, const std::vector<int32>& strides, in ConvBackpropExtractAndVerifyDimension()
98 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides, in ConvBackpropComputeDimensionsV2()
|
| D | conv_grad_filter_ops.cc | 118 std::vector<int32> dilations(4, 1); in operator ()() local
679 std::vector<int32> dilations(4, 1); in operator ()() local
|
| D | conv_grad_input_ops.cc | 84 std::vector<int32> dilations(4, 1); in operator ()() local
|
| D | depthwise_conv_op.cc | 306 std::vector<int32_t> dilations; in DepthwiseConv2dNativeOp() local
|
| D | conv_ops_3d.cc | 172 std::array<int64_t, 3> dilations = { in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.h | 51 std::vector<int32> dilations; member
|
| D | conv_op_helpers.cc | 139 absl::Span<const int32> dilations, const std::vector<int32>& strides, in ConvBackpropComputeDimensionsV2XlaShapes()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | conv_node_name_test.py | 38 dilations): argument
|
| D | conv2d_test.py | 44 dilations=None, argument
239 dilations=None, argument
553 dilations=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
| D | tf_to_xla_attribute_utils.cc | 91 int8_t input_zp_value, ArrayAttr strides, ArrayAttr dilations, in PadForDynamicShapedInputSamePadding()
139 int8_t input_zp_value, ArrayAttr strides, ArrayAttr dilations, in CalculatePaddingAndPadIfNeeded()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 92 virtual inline void GetDilationsInMklOrder(memory::dims* dilations) { in GetDilationsInMklOrder()
541 const memory::dims& dilations, memory::dims* output_dims_tf_order, in GetOutputAndPadSizeInMklOrder()
|
| D | mkl_conv_grad_input_ops.cc | 54 memory::dims dilations; member
370 memory::dims padding_left, padding_right, dilations, strides; in Compute() local
|
| D | mkl_conv_grad_filter_ops.cc | 50 memory::dims dilations; member
421 memory::dims padding_left, padding_right, dilations, strides; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 96 const ArrayAttr& dilations = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | kernel_shape_util.cc | 122 const std::array<int64_t, 3>& dilations, in Get3dOutputSizeV2()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.proto | 158 repeated int64 dilations = 3; field
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 2618 std::vector<int32> dilations; in CopyAttrsConv() local
2653 std::vector<int32> dilations; in CopyAttrsFromPadAndConv2D() local
2685 std::vector<int32> dilations; in CopyAttrsFromPadAndFusedConv2D() local
2723 std::vector<int32> strides, dilations, padding_list; in CopyAttrsQuantizedConv2D() local
2795 const std::vector<int32>& dilations, bool change_format) { in CopyFormatAttrsConv()
2950 std::vector<int32> dilations; in MergeConv2DWithBiasAdd() local
3116 std::vector<int32> dilations; in MergePadWithConv2D() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad.cc | 523 std::vector<int> dilations, strides, explicit_paddings; in Conv2DBackpropInputGrad() local
569 std::vector<int> dilations, strides, explicit_paddings; in DepthwiseConv2dNativeGrad() local
|