Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py243 dilations, argument
312 dilations=None): argument
769 dilations=None): argument
903 dilations=None, argument
927 dilations=None, argument
1091 dilations=None): argument
1263 dilations=None, argument
1549 dilations=None): argument
1634 dilations=None, argument
1700 dilations=None, argument
[all …]
Dnn_impl.py511 dilations=None): argument
585 dilations=None, argument
649 dilations=None): argument
749 dilations=None, argument
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_ops.cc60 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides, in ConvBackpropExtractAndVerifyDimension()
101 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides, in ConvBackpropComputeDimensionsV2()
Dmkl_conv_ops.h88 virtual inline void GetDilationsInMklOrder(memory::dims* dilations) { in GetDilationsInMklOrder()
478 const memory::dims& dilations, memory::dims* output_dims_tf_order, in GetOutputAndPadSizeInMklOrder()
Dmkl_conv_grad_input_ops.cc59 memory::dims dilations; member
362 memory::dims padding_left, padding_right, dilations, strides; in Compute() local
Dmkl_conv_grad_filter_ops.cc57 memory::dims dilations; member
425 memory::dims padding_left, padding_right, dilations, strides, in Compute() local
Dconv_ops_3d.cc148 std::array<int64, 3> dilations = { in Compute() local
Dquantized_conv_ops.cc466 std::vector<int32> dilations; in QuantizedConv2DOp() local
Dconv_grad_filter_ops.cc553 std::vector<int32> dilations(4, 1); in operator ()() local
Dmkl_conv_ops.cc72 memory::dims dilations; member
941 dilations, strides; in Compute() local
Dconv_grad_input_ops.cc691 std::vector<int32> dilations(4, 1); in operator ()() local
Dconv_ops.cc295 const auto& dilations = params->dilations; in InitConv2DParameters() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.h47 std::vector<int32> dilations; member
Dconv_op_helpers.cc205 absl::Span<const int32> dilations, const std::vector<int32>& strides, in ConvBackpropComputeDimensionsV2XlaShapes()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py179 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, dilations, argument
300 padding, dilations, rtol=1e-4): argument
324 dilations=(1, 1), argument
367 dilations=(1, 1), argument
713 dilations=(1, 1)): argument
901 dilations=(1, 1), argument
1075 output_sizes, strides, dilations, argument
1121 output_sizes, strides, dilations, argument
1323 dilations=(1, 1), argument
1484 dilations=(1, 1), argument
Dconv_ops_3d_test.py170 padding, dilations): argument
684 strides, dilations, padding, data_format, use_gpu, argument
/external/tensorflow/tensorflow/compiler/tests/
Dconv2d_test.py48 dilations=None, argument
243 dilations=None, argument
541 dilations=None, argument
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc2044 std::vector<int32> dilations; in CopyAttrsConvCheckConstFilter() local
2069 std::vector<int32> dilations; in CopyAttrsConv() local
2094 std::vector<int32> dilations; in CopyAttrsPadWithConv2D() local
2149 std::vector<int32> dilations; in CopyAttrsFromPadAndConv2D() local
2181 std::vector<int32> dilations; in CopyAttrsFromPadAndFusedConv2D() local
2216 std::vector<int32> dilations; in CopyAttrsConv2DDepthwise() local
2239 std::vector<int32> dilations; in CopyAttrsConv2DDepthwiseCheckConstFilter() local
2391 std::vector<int32> strides, dilations, padding_list; in CopyAttrsQuantizedConv2D() local
2490 const std::vector<int32>& dilations, bool change_format) { in CopyFormatAttrsConv()
2590 std::vector<int32> dilations; in CopyAttrsFusedConv2D() local
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto102 repeated int64 dilations = 3; field
Ddnn.cc471 string dilations; in ToString() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns_test.cc600 const std::vector<int32>& explicit_paddings = {}) { in TEST()
799 const string& padding) { in TEST()
Dcommon_shape_fns.cc121 const std::array<int64, 3>& dilations, in Get3dOutputSizeV2()
444 std::vector<int32> dilations; in Conv2DShapeImpl() local
565 std::vector<int32> dilations; in Conv3DShape() local
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc662 const auto& dilations = GetListAttr(node, "dilations"); in ConvertConvOperator() local
735 const auto& dilations = GetListAttr(node, "dilations"); in ConvertDepthwiseConvOperator() local
1872 const auto& dilations = GetListAttr(node, "dilations"); in ConvertTransposeConvOperator() local
Dexport_tensorflow.cc411 auto& dilations = (*conv2d_op->mutable_attr())["dilations"]; in ConvertConvOperator() local
502 auto& dilations = (*dc2d_op->mutable_attr())["dilations"]; in ConvertDepthwiseConvOperator() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc3114 bool is_conv2d_backprop_input = false) -> NodeDef { in TEST_F()
3244 std::vector<int> dilations; in TEST_F() member

12