| /external/tensorflow/tensorflow/core/kernels/ |
| avgpooling_op.cc | 62 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
63 OP_REQUIRES(context, ksize_.size() == 4,
71 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
78 PoolParameters params{context, ksize_, stride_,
99 std::vector<int32> ksize_; member in class:tensorflow::AvgPoolingOp
122 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
123 OP_REQUIRES(context, ksize_.size() == 4,
131 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N');
140 PoolParameters params{context, ksize_, stride_,
157 context, perftools::gputools::dnn::PoolingMode::kAverage, ksize_,
173 std::vector<int32> ksize_; member in class:tensorflow::AvgPoolingOp
347 std::vector<int32> ksize_; member in class:tensorflow::AvgPoolingGradOp
420 std::vector<int32> ksize_; member in class:tensorflow::AvgPoolingGradOp
545 std::vector<int32> ksize_; member in class:tensorflow::AvgPoolingGradOpCustomGPUKernel
[all...] |
| maxpooling_op.cc | 212 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
213 OP_REQUIRES(context, ksize_.size() == 4,
220 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
224 context, ksize_[3] == 1 && stride_[3] == 1,
256 std::vector<int32> ksize = ksize_;
300 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingGradOp
343 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
344 OP_REQUIRES(context, ksize_.size() == 4,
351 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N');
380 std::vector<int32> ksize = ksize_;
419 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingGradOp
630 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingGradGradOp
726 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingGradGradOp
785 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingNoMaskOp
863 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingNoMaskV2Op
915 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingWithArgmaxOp
972 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingGradWithArgmaxOp
1023 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingGradGradWithArgmaxOp
1106 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingNoMaskOp
1199 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingNoMaskV2Op
[all...] |
| quantized_pooling_ops.cc | 41 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
42 OP_REQUIRES(context, ksize_.size() == 4,
50 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
57 PoolParameters params{context, ksize_, stride_,
103 std::vector<int32> ksize_; member in class:tensorflow::QuantizedAvgPoolingOp
|
| pooling_ops_3d.cc | 145 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
146 OP_REQUIRES(context, ksize_.size() == 5,
155 (GetTensorDim(ksize_, data_format_, 'N') == 1 &&
160 (GetTensorDim(ksize_, data_format_, 'C') == 1 &&
179 std::array<int64, 3> window{{GetTensorDim(ksize_, data_format_, '2'),
180 GetTensorDim(ksize_, data_format_, '1'),
181 GetTensorDim(ksize_, data_format_, '0')}};
200 std::vector<int32> ksize_; member in class:tensorflow::Pooling3DOp
318 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
319 OP_REQUIRES(context, ksize_.size() == 5
374 std::vector<int32> ksize_; member in class:tensorflow::MaxPooling3dGradOp
535 std::vector<int32> ksize_; member in class:tensorflow::AvgPooling3dGradOp
710 std::vector<int32> ksize_; member in class:tensorflow::MaxPooling3dGradGradOp
[all...] |
| mkl_pooling_ops_common.h | 118 OP_REQUIRES_OK(context, context->GetAttr("ksize", &this->ksize_));
119 OP_REQUIRES(context, this->ksize_.size() == 4,
127 OP_REQUIRES(context, this->ksize_[0] == 1 && this->stride_[0] == 1,
158 pool_params->Init(context, this->ksize_, this->stride_, this->padding_,
161 pool_params->Init(context, this->ksize_, this->stride_, this->padding_,
178 std::vector<int32> ksize_; member in class:tensorflow::MklPoolingOpBase
|
| mkl_maxpooling_op.cc | 55 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
56 OP_REQUIRES(context, ksize_.size() == 4,
64 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
85 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
92 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
190 std::vector<int32> ksize_; member in class:tensorflow::MklMaxPoolingOp
213 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
214 OP_REQUIRES(context, ksize_.size() == 4,
222 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
251 pool_params.Init(context, ksize_, stride_, padding_, data_format_
489 std::vector<int32> ksize_; member in class:tensorflow::MklMaxPoolingGradOp
[all...] |
| mkl_avgpooling_op.cc | 54 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
55 OP_REQUIRES(context, ksize_.size() == 4,
63 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
81 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
84 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
211 std::vector<int32> ksize_; member in class:tensorflow::MklAvgPoolingOp
229 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
230 OP_REQUIRES(context, ksize_.size() == 4,
238 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
258 pool_params.Init(context, ksize_, stride_, padding_, data_format_
425 std::vector<int32> ksize_; member in class:tensorflow::MklAvgPoolingGradOp
[all...] |
| pooling_ops_common.h | 95 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
96 OP_REQUIRES(context, ksize_.size() == 4,
104 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
111 PoolParameters params{context, ksize_, stride_,
258 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingOp
307 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
308 OP_REQUIRES(context, ksize_.size() == 4,
315 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
325 std::vector<int32> ksize = ksize_;
505 std::vector<int32> ksize_; member in class:tensorflow::MaxPoolingV2Op
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| pooling_ops.cc | 54 ksize_.push_back(ksize_int[i]);
82 std::vector<int64> ksize = ksize_;
123 std::vector<int64> ksize_; member in class:tensorflow::__anon38761::PoolingOp
245 ksize_, stride_, num_spatial_dims_,
280 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_));
298 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_));
308 OP_REQUIRES(ctx, ksize_.size() == num_dims(),
348 input, select, ksize_, stride_, xla_padding, out_backprop, init_value,
356 std::vector<int64> ksize_; member in class:tensorflow::__anon38761::MaxPoolGradOp
390 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_));
501 std::vector<int64> ksize_; member in class:tensorflow::__anon38761::AvgPoolGradOp
[all...] |
