Home
last modified time | relevance | path

Searched defs:dilations (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py247 dilations, argument
316 dilations=None): argument
773 dilations=None): argument
907 dilations=None, argument
931 dilations=None, argument
1119 dilations=None): argument
1291 dilations=None, argument
1589 dilations=None): argument
1674 dilations=None, argument
1740 dilations=None, argument
[all …]
Dnn_impl.py719 dilations=None): argument
808 dilations=None, argument
872 dilations=None): argument
972 dilations=None, argument
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc55 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides, in ConvBackpropExtractAndVerifyDimension()
96 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides, in ConvBackpropComputeDimensionsV2()
Dmkl_conv_ops.h98 virtual inline void GetDilationsInMklOrder(memory::dims* dilations) { in GetDilationsInMklOrder()
491 const memory::dims& dilations, memory::dims* output_dims_tf_order, in GetOutputAndPadSizeInMklOrder()
Dmkl_conv_grad_input_ops.cc61 memory::dims dilations; member
366 memory::dims padding_left, padding_right, dilations, strides; in Compute() local
Dmkl_conv_grad_filter_ops.cc57 memory::dims dilations; member
428 memory::dims padding_left, padding_right, dilations, strides, in Compute() local
Dconv_grad_filter_ops.cc114 std::vector<int32> dilations(4, 1); in operator ()() local
661 std::vector<int32> dilations(4, 1); in operator ()() local
Dconv_ops_3d.cc155 std::array<int64, 3> dilations = { in Compute() local
Dconv_grad_input_ops.cc118 std::vector<int32> dilations(4, 1); in operator ()() local
841 std::vector<int32> dilations(4, 1); in operator ()() local
Dquantized_conv_ops.cc466 std::vector<int32> dilations; in QuantizedConv2DOp() local
Dmkl_conv_ops.cc67 memory::dims dilations; member
528 dilations, strides; in Compute() local
Dconv_ops.cc361 const auto& dilations = params->dilations; in InitConv2DParameters() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.h51 std::vector<int32> dilations; member
Dconv_op_helpers.cc163 absl::Span<const int32> dilations, const std::vector<int32>& strides, in ConvBackpropComputeDimensionsV2XlaShapes()
/external/tensorflow/tensorflow/compiler/tests/
Dconv_node_name_test.py42 dilations): argument
Dconv2d_test.py48 dilations=None, argument
243 dilations=None, argument
541 dilations=None, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py182 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
706 def _VerifyGroupConvFwd(self, tensor_in_sizes, filter_in_sizes, dilations, argument
832 dilations=(1, 1)): argument
1020 dilations=(1, 1), argument
1194 output_sizes, strides, dilations, argument
1240 output_sizes, strides, dilations, argument
1443 dilations=(1, 1), argument
[all …]
Dconv_ops_3d_test.py174 padding, dilations): argument
721 strides, dilations, padding, data_format, use_gpu, argument
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc2477 std::vector<int32> dilations; in CopyAttrsConvCheckConstFilter() local
2502 std::vector<int32> dilations; in CopyAttrsConv() local
2527 std::vector<int32> dilations; in CopyAttrsPadWithConv2D() local
2582 std::vector<int32> dilations; in CopyAttrsFromPadAndConv2D() local
2614 std::vector<int32> dilations; in CopyAttrsFromPadAndFusedConv2D() local
2648 std::vector<int32> dilations; in CopyAttrsConv2DDepthwiseCheckConstFilter() local
2675 std::vector<int32> strides, dilations, padding_list; in CopyAttrsQuantizedConv2D() local
2739 const std::vector<int32>& dilations, bool change_format) { in CopyFormatAttrsConv()
2785 std::vector<int32> dilations; in CopyAttrsFusedConv2D() local
2931 std::vector<int32> dilations; in MergeConv2DWithBiasAdd() local
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto105 repeated int64 dilations = 3; field
Ddnn.cc497 string dilations; in ToString() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h221 HW dilations = HW(1, 1); // Along each axis. member
230 HWD dilations = HWD(0, 0, 0); // Along each axis. member
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns_test.cc818 const std::vector<int32>& explicit_paddings = {}) { in TEST()
1034 const string& padding) { in TEST()
Dcommon_shape_fns.cc130 const std::array<int64, 3>& dilations, in Get3dOutputSizeV2()
699 std::vector<int32> dilations; in Conv2DShapeImpl() local
836 std::vector<int32> dilations; in Conv3DShape() local
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc831 const auto& dilations = GetListAttr(node, "dilations"); in ConvertConvOperator() local
910 const auto& dilations = GetListAttr(node, "dilations"); in ConvertDepthwiseConvOperator() local
1952 const auto& dilations = GetListAttr(node, "dilations"); in ConvertTransposeConvOperator() local

12