| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_ops_3d.cc | 183 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilation_));
184 OP_REQUIRES(context, dilation_.size() == 5,
188 (GetTensorDim(dilation_, data_format_, 'C') == 1 &&
189 GetTensorDim(dilation_, data_format_, 'N') == 1),
196 (GetTensorDim(dilation_, data_format_, '0') == 1 &&
197 GetTensorDim(dilation_, data_format_, '1') == 1 &&
198 GetTensorDim(dilation_, data_format_, '2') == 1),
253 std::vector<int32> dilation_; member in class:tensorflow::Conv3DBackpropInputOp
289 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilation_));
290 OP_REQUIRES(context, dilation_.size() == 5
573 std::vector<int32> dilation_; member in class:tensorflow::Conv3DCustomBackpropInputOp
715 std::vector<int32> dilation_; member in class:tensorflow::Conv3DBackpropFilterOp
993 std::vector<int32> dilation_; member in class:tensorflow::Conv3DCustomBackpropFilterOp
1456 std::vector<int32> dilation_; member in class:tensorflow::Conv3DBackpropInputOp
1839 std::vector<int32> dilation_; member in class:tensorflow::Conv3DBackpropFilterOp
[all...] |
| conv_ops_3d.cc | 97 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilation_));
98 OP_REQUIRES(context, dilation_.size() == 5,
102 (GetTensorDim(dilation_, data_format_, 'N') == 1 &&
103 GetTensorDim(dilation_, data_format_, 'C') == 1),
109 (GetTensorDim(dilation_, data_format_, '0') > 0 &&
110 GetTensorDim(dilation_, data_format_, '1') > 0 &&
111 GetTensorDim(dilation_, data_format_, '2') > 0),
149 {GetTensorDim(dilation_, data_format_, '0'),
150 GetTensorDim(dilation_, data_format_, '1'),
151 GetTensorDim(dilation_, data_format_, '2')}}
174 std::vector<int32> dilation_; member in class:tensorflow::Conv3DOp
[all...] |
| conv_ops_gpu.h | 98 dilation_(CheckSpatialArraySize(dilation)),
132 "(", str_util::Join(dilation_, ", "), "), ",
164 filter_, dilation_, stride_, padding_, dtype_,
197 SpatialArray dilation_; member in class:tensorflow::ConvParameters
|
