| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops_3d.cc | 192 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilation_)); in Conv3DBackpropInputOp()
193 OP_REQUIRES(context, dilation_.size() == 5, in Conv3DBackpropInputOp()
197 (GetTensorDim(dilation_, data_format_, 'C') == 1 && in Conv3DBackpropInputOp()
198 GetTensorDim(dilation_, data_format_, 'N') == 1), in Conv3DBackpropInputOp()
205 (GetTensorDim(dilation_, data_format_, '0') == 1 && in Conv3DBackpropInputOp()
206 GetTensorDim(dilation_, data_format_, '1') == 1 && in Conv3DBackpropInputOp()
207 GetTensorDim(dilation_, data_format_, '2') == 1), in Conv3DBackpropInputOp()
263 std::vector<int32> dilation_; member in tensorflow::Conv3DBackpropInputOp
299 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilation_)); in Conv3DCustomBackpropInputOp()
300 OP_REQUIRES(context, dilation_.size() == 5, in Conv3DCustomBackpropInputOp()
[all …]
|
| D | conv_ops_gpu.h | 114 dilation_(CheckSpatialArraySize(dilation)), in batch_()
151 "(", str_util::Join(dilation_, ", "), "), ", in ToString()
184 filter_, dilation_, stride_, padding_, dtype_, in get_data_as_tuple()
217 SpatialArray dilation_; variable
|
| D | conv_ops_3d.cc | 103 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilation_)); in Conv3DOp()
104 OP_REQUIRES(context, dilation_.size() == 5, in Conv3DOp()
108 (GetTensorDim(dilation_, data_format_, 'N') == 1 && in Conv3DOp()
109 GetTensorDim(dilation_, data_format_, 'C') == 1), in Conv3DOp()
115 (GetTensorDim(dilation_, data_format_, '0') > 0 && in Conv3DOp()
116 GetTensorDim(dilation_, data_format_, '1') > 0 && in Conv3DOp()
117 GetTensorDim(dilation_, data_format_, '2') > 0), in Conv3DOp()
158 {GetTensorDim(dilation_, data_format_, '0'), in Compute()
159 GetTensorDim(dilation_, data_format_, '1'), in Compute()
160 GetTensorDim(dilation_, data_format_, '2')}}; in Compute()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_powervr.cc | 177 dilation_(attr.dilations.w, attr.dilations.h, 1, 1), in ConvPowerVR()
188 dilation_(attr.dilations.w, attr.dilations.h, 1, 1), in ConvPowerVR()
199 dilation_(1, 1, 1, 1), in ConvPowerVR()
207 dilation_(1, 1, 1, 1) {} in ConvPowerVR()
214 dilation_(operation.dilation_), in ConvPowerVR()
226 dilation_(attr.dilations.w, attr.dilations.h, attr.dilations.d, 1), in ConvPowerVR()
234 std::swap(dilation_, operation.dilation_); in operator =()
274 RETURN_IF_ERROR(args->SetInt("dilation_x", dilation_.x * src_[0]->Batch())); in BindArguments()
280 RETURN_IF_ERROR(args->SetInt("dilation_y", dilation_.y)); in BindArguments()
287 RETURN_IF_ERROR(args->SetInt("dilation_z", dilation_.z)); in BindArguments()
|
| D | conv_metal.h | 91 int2 dilation_; variable
|
| D | conv_metal.cc | 959 RETURN_IF_ERROR(args->SetInt("dilation_x", dilation_.x * src_[0]->Batch())); in BindArguments()
1020 desc.dilation_ = int2(attr.dilations.w, attr.dilations.h); in CreateConvolutionMetal()
1145 desc.dilation_ = int2(1, 1); in CreateConvolutionMetalWino4x4To6x6()
|
| D | conv_powervr.h | 201 int4 dilation_; variable
|