Home
last modified time | relevance | path

Searched refs:dilation_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_ops_3d.cc183 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilation_)); in Conv3DBackpropInputOp()
184 OP_REQUIRES(context, dilation_.size() == 5, in Conv3DBackpropInputOp()
188 (GetTensorDim(dilation_, data_format_, 'C') == 1 && in Conv3DBackpropInputOp()
189 GetTensorDim(dilation_, data_format_, 'N') == 1), in Conv3DBackpropInputOp()
196 (GetTensorDim(dilation_, data_format_, '0') == 1 && in Conv3DBackpropInputOp()
197 GetTensorDim(dilation_, data_format_, '1') == 1 && in Conv3DBackpropInputOp()
198 GetTensorDim(dilation_, data_format_, '2') == 1), in Conv3DBackpropInputOp()
253 std::vector<int32> dilation_; member in tensorflow::Conv3DBackpropInputOp
289 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilation_)); in Conv3DCustomBackpropInputOp()
290 OP_REQUIRES(context, dilation_.size() == 5, in Conv3DCustomBackpropInputOp()
[all …]
Dconv_ops_3d.cc97 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilation_)); in Conv3DOp()
98 OP_REQUIRES(context, dilation_.size() == 5, in Conv3DOp()
102 (GetTensorDim(dilation_, data_format_, 'N') == 1 && in Conv3DOp()
103 GetTensorDim(dilation_, data_format_, 'C') == 1), in Conv3DOp()
109 (GetTensorDim(dilation_, data_format_, '0') > 0 && in Conv3DOp()
110 GetTensorDim(dilation_, data_format_, '1') > 0 && in Conv3DOp()
111 GetTensorDim(dilation_, data_format_, '2') > 0), in Conv3DOp()
149 {GetTensorDim(dilation_, data_format_, '0'), in Compute()
150 GetTensorDim(dilation_, data_format_, '1'), in Compute()
151 GetTensorDim(dilation_, data_format_, '2')}}; in Compute()
[all …]
Dconv_ops_gpu.h98 dilation_(CheckSpatialArraySize(dilation)), in ConvParameters()
132 "(", str_util::Join(dilation_, ", "), "), ", in ToString()
164 filter_, dilation_, stride_, padding_, dtype_, in get_data_as_tuple()
197 SpatialArray dilation_; variable