| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_maxpooling_op.cc | 53 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
85 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
92 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
98 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params);
111 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_);
193 TensorFormat data_format_; member in class:tensorflow::MklMaxPoolingOp
211 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
251 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
258 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
263 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params)
492 TensorFormat data_format_; member in class:tensorflow::MklMaxPoolingGradOp
[all...] |
| conv_grad_filter_ops.cc | 185 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
187 OP_REQUIRES(context, data_format_ == FORMAT_NHWC,
235 out_backprop.shape(), strides_, padding_, data_format_, &dims));
270 data_format_)) {
278 dims.spatial_dims[1].stride, padding_, filter_backprop, data_format_);
285 TensorFormat data_format_; member in class:tensorflow::Conv2DFastBackpropFilterOp
298 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
300 OP_REQUIRES(context, data_format_ == FORMAT_NHWC,
349 strides_, padding_, data_format_, &dims));
383 data_format_)) {
499 TensorFormat data_format_; member in class:tensorflow::Conv2DCustomBackpropFilterOp
623 TensorFormat data_format_; member in class:tensorflow::Conv2DSlowBackpropFilterOp
[all...] |
| conv_grad_input_ops.cc | 188 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
190 OP_REQUIRES(context, data_format_ == FORMAT_NHWC,
238 strides_, padding_, data_format_, &dims));
274 data_format_)) {
283 in_backprop, data_format_);
290 TensorFormat data_format_; member in class:tensorflow::Conv2DFastBackpropInputOp
303 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
305 OP_REQUIRES(context, data_format_ == FORMAT_NHWC,
353 strides_, padding_, data_format_, &dims));
391 data_format_)) {
571 TensorFormat data_format_; member in class:tensorflow::Conv2DCustomBackpropInputOp
692 TensorFormat data_format_; member in class:tensorflow::Conv2DSlowBackpropInputOp
[all...] |
| mkl_conv_ops.cc | 74 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
80 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
81 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
127 : GetTensorDim(input, data_format_, 'C');
139 : GetTensorDim(input, data_format_, 'H');
151 : GetTensorDim(input, data_format_, 'W');
162 : GetTensorDim(input, data_format_, 'N');
171 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
172 const int stride_cols = GetTensorDim(strides_, data_format_, 'W');
182 ShapeFromFormat(data_format_, batch, out_rows, out_cols, out_depth)
483 TensorFormat data_format_; member in class:tensorflow::MklConv2DOp
660 TensorFormat data_format_; member in class:tensorflow::MklConv2DOp
[all...] |
