| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 59 std::vector<int32> strides_; variable
69 strides_(strides), in context_()
81 if (strides_.size() == 4) { in GetStridesInMklOrder()
82 int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in GetStridesInMklOrder()
83 int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in GetStridesInMklOrder()
85 } else if (strides_.size() == 5) { in GetStridesInMklOrder()
86 int stride_planes = GetTensorDim(strides_, data_format_, '0'); in GetStridesInMklOrder()
87 int stride_rows = GetTensorDim(strides_, data_format_, '1'); in GetStridesInMklOrder()
88 int stride_cols = GetTensorDim(strides_, data_format_, '2'); in GetStridesInMklOrder()
134 if (strides_.size() == 4) { // NCHW format for Conv2D in GetInputSizeInMklOrder()
[all …]
|
| D | mkl_conv_ops.cc | 440 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in MklConvOp()
445 OP_REQUIRES(context, (strides_.size() == 4 || strides_.size() == 5), in MklConvOp()
449 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvOp()
450 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp()
463 if (strides_.size() == 4) { in MklConvOp()
478 } else if (strides_.size() == 5) { in MklConvOp()
535 MklDnnConvUtil conv_utl(context, strides_, padding_, data_format_, in Compute()
575 bool is_conv2d = (strides_.size() == 4); in Compute()
938 std::vector<int32> strides_; member in tensorflow::MklConvOp
|
| D | mkl_conv_grad_filter_ops.cc | 423 MklDnnConvUtil conv_util(context, this->strides_, this->padding_, in Compute()
431 bool is_conv2d = (this->strides_.size() == 4); in Compute()
670 : ((this->strides_.size() == 4) ? memory::format_tag::hwio in GetOutputFormat()
|
| D | mkl_conv_grad_input_ops.cc | 375 MklDnnConvUtil conv_util(context, this->strides_, this->padding_, in Compute()
383 bool is_conv2d = (this->strides_.size() == 4); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | neon_depthwise_conv_op.cc | 46 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in NeonDepthwiseConv2dNativeOp()
47 OP_REQUIRES(context, strides_.size() == 4, in NeonDepthwiseConv2dNativeOp()
50 OP_REQUIRES(context, strides_[1] == strides_[2], in NeonDepthwiseConv2dNativeOp()
55 context, (strides_[0] == 1 && strides_[3] == 1), in NeonDepthwiseConv2dNativeOp()
88 const int32 stride = strides_[1]; in Compute()
189 std::vector<int32> strides_; member in tensorflow::NeonDepthwiseConv2dNativeOp
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 38 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &strides_)); in ExtractImagePatchesOp()
52 OP_REQUIRES(ctx, strides_.size() == num_dims, in Compile()
69 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1, in Compile()
84 OP_REQUIRES(ctx, strides_[input_dim] >= 1, in Compile()
144 window_strides[i] = strides_.at(dim); in Compile()
174 std::vector<int32> strides_; member in tensorflow::__anon1d3b63380111::ExtractImagePatchesOp
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_volume_patches_op.cc | 65 ParseAttributeVec5(context, "strides", &strides_); in ExtractVolumePatchesOp()
88 const int stride_planes = strides_[1]; in Compute()
89 const int stride_rows = strides_[2]; in Compute()
90 const int stride_cols = strides_[3]; in Compute()
146 std::vector<int32> strides_; member in tensorflow::ExtractVolumePatchesOp
|
| D | extract_image_patches_op.cc | 59 ParseAttributeVec4(context, "strides", &strides_); in ExtractImagePatchesOp()
80 const int stride_rows = strides_[1]; in Compute()
81 const int stride_cols = strides_[2]; in Compute()
118 std::vector<int32> strides_; member in tensorflow::ExtractImagePatchesOp
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.h | 362 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
363 OP_REQUIRES(context, strides_.size() == 4,
366 int stride_n = GetTensorDim(strides_, data_format_, 'N');
367 int stride_c = GetTensorDim(strides_, data_format_, 'C');
368 int stride_h = GetTensorDim(strides_, data_format_, 'H');
369 int stride_w = GetTensorDim(strides_, data_format_, 'W');
441 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
442 const int stride_cols = GetTensorDim(strides_, data_format_, 'W');
461 std::vector<int32> strides_;
485 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
[all …]
|
| D | dilation_ops.cc | 118 ParseAttributes(context, &strides_, &rates_, &padding_); in DilationOp()
130 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, in Compute()
155 std::vector<int32> strides_; member in tensorflow::DilationOp
216 ParseAttributes(context, &strides_, &rates_, &padding_); in DilationBackpropInputOp()
229 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, in Compute()
262 std::vector<int32> strides_; member in tensorflow::DilationBackpropInputOp
336 ParseAttributes(context, &strides_, &rates_, &padding_); in DilationBackpropFilterOp()
349 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, in Compute()
382 std::vector<int32> strides_; member in tensorflow::DilationBackpropFilterOp
|
| D | conv_ops_using_gemm.cc | 435 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in Conv2DUsingGemmOp()
443 OP_REQUIRES(context, strides_.size() == 4, in Conv2DUsingGemmOp()
446 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DUsingGemmOp()
447 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DUsingGemmOp()
519 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute()
520 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute()
558 std::vector<int32> strides_; member in tensorflow::Conv2DUsingGemmOp
|
| D | quantized_conv_ops.cc | 454 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in QuantizedConv2DOp()
455 OP_REQUIRES(context, strides_.size() == 4, in QuantizedConv2DOp()
458 OP_REQUIRES(context, strides_[1] == strides_[2], in QuantizedConv2DOp()
463 context, (strides_[0] == 1 && strides_[3] == 1), in QuantizedConv2DOp()
538 const int stride = strides_[1]; in Compute()
583 std::vector<int32> strides_; member in tensorflow::QuantizedConv2DOp
|
| D | conv_grad_filter_ops.cc | 269 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in Conv2DBackpropFilterOp()
270 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DBackpropFilterOp()
271 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DBackpropFilterOp()
272 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DBackpropFilterOp()
273 int stride_w = GetTensorDim(strides_, data_format_, 'W'); in Conv2DBackpropFilterOp()
349 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute()
350 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute()
368 std::vector<int32> strides_; member in tensorflow::Conv2DBackpropFilterOp
392 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in Conv2DCustomBackpropFilterOp()
393 OP_REQUIRES(context, strides_.size() == 4, in Conv2DCustomBackpropFilterOp()
[all …]
|
| D | depthwise_conv_op.cc | 273 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in DepthwiseConv2dNativeOp()
279 OP_REQUIRES(context, strides_.size() == 4, in DepthwiseConv2dNativeOp()
282 stride_ = GetTensorDim(strides_, data_format_, 'H'); in DepthwiseConv2dNativeOp()
283 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeOp()
284 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeOp()
285 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeOp()
477 std::vector<int32> strides_; member in tensorflow::DepthwiseConv2dNativeOp
|
| D | depthwise_conv_grad_op.cc | 553 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in DepthwiseConv2dNativeBackpropInputOp()
554 OP_REQUIRES(context, strides_.size() == 4, in DepthwiseConv2dNativeBackpropInputOp()
563 stride_ = GetTensorDim(strides_, data_format_, 'H'); in DepthwiseConv2dNativeBackpropInputOp()
564 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropInputOp()
565 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeBackpropInputOp()
566 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropInputOp()
694 std::vector<int32> strides_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp
1056 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in DepthwiseConv2dNativeBackpropFilterOp()
1057 OP_REQUIRES(context, strides_.size() == 4, in DepthwiseConv2dNativeBackpropFilterOp()
1066 stride_ = GetTensorDim(strides_, data_format_, 'H'); in DepthwiseConv2dNativeBackpropFilterOp()
[all …]
|
| D | conv_ops_fused_image_transform.cc | 633 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in FusedResizeConv2DUsingGemmOp()
634 OP_REQUIRES(context, strides_.size() == 4, in FusedResizeConv2DUsingGemmOp()
637 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N'); in FusedResizeConv2DUsingGemmOp()
638 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); in FusedResizeConv2DUsingGemmOp()
819 const int stride_rows = GetTensorDim(strides_, FORMAT_NHWC, 'H'); in Compute()
820 const int stride_cols = GetTensorDim(strides_, FORMAT_NHWC, 'W'); in Compute()
863 std::vector<int32> strides_; member in tensorflow::FusedResizeConv2DUsingGemmOp
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | strided_slice_test.cc | 35 strides_ = AddConstInput(strides, strides_data); in StridedSliceOpModel()
44 GetShape(strides_)}); in StridedSliceOpModel()
60 int strides_; member in tflite::StridedSliceOpModel
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 538 strides_(ndims, 1) {} in PoolingDescriptor()
547 strides_ = other.strides_; in CloneFrom()
558 absl::StrAppendFormat(&strides, "%d ", strides_[i]); in ToString()
573 absl::StrAppendFormat(&strides, "_s%d:%d", i, strides_[i]); in ToShortString()
|
| D | dnn.h | 687 SetDim(&strides_, DimIndex::Y, value); in set_vertical_stride()
691 SetDim(&strides_, DimIndex::X, value); in set_horizontal_stride()
695 SetDim(&strides_, dim, value); in set_stride()
720 int64 vertical_stride() const { return GetDim(strides_, DimIndex::Y); } in vertical_stride()
721 int64 horizontal_stride() const { return GetDim(strides_, DimIndex::X); } in horizontal_stride()
722 int64 stride(DimIndex dim) const { return GetDim(strides_, dim); } in stride()
725 absl::Span<const int64> strides() const { return strides_; } in strides()
738 std::vector<int64> strides_; variable
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 42 strides_ = AddInput(TensorType_INT32); in StridedSliceOpModel()
68 PopulateTensor<int32_t>(strides_, data); in SetStrides()
83 int strides_; member in tflite::__anone1578cc60111::StridedSliceOpModel
|
| /external/python/pybind11/include/pybind11/ |
| D | numpy.h | 359 shape_, strides_;
369 strides_[i] = strides[i];
375 …: data_{reinterpret_cast<const unsigned char *>(data)}, shape_{shape}, strides_{strides}, dims_{di…
386 … return *reinterpret_cast<const T *>(data_ + byte_offset_unsafe(strides_, ssize_t(index)...));
|
| /external/libaom/libaom/test/ |
| D | noise_model_test.cc | 534 strides_[c] = kWidth; in SetUp()
545 kWidth, kHeight, strides_, chroma_sub_, in NoiseModelUpdate()
564 int strides_[3]; member in NoiseModelUpdateTest
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2522 strides_ = AddConstInput(TensorType_INT32, strides_data, strides_shape); in StridedSliceOpModel()
2546 int strides_; member in tflite::__anon0c68fcb70111::StridedSliceOpModel
|