/external/tensorflow/tensorflow/contrib/lite/kernels/ |
strided_slice_test.cc | 38 strides_ = AddInput(TensorType_INT32); 58 PopulateTensor<int32>(strides_, data); 68 int strides_; member in class:tflite::__anon39279::StridedSliceOpModel
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
extract_image_patches_op.cc | 30 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &strides_)); 44 OP_REQUIRES(ctx, strides_.size() == num_dims, 61 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1, 76 OP_REQUIRES(ctx, strides_[input_dim] >= 1, 139 window_strides[i] = strides_.at(dim); 159 std::vector<int32> strides_; member in class:tensorflow::__anon38742::ExtractImagePatchesOp
|
conv_ops.cc | 183 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &strides_)); 195 OP_REQUIRES(ctx, strides_.size() == num_dims(), 202 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1, 276 window_strides[i] = strides_.at(dim); 297 std::vector<int32> strides_; member in class:tensorflow::__anon38736::ConvOp 335 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &strides_)); 346 OP_REQUIRES(ctx, strides_.size() == num_dims(), 353 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1 450 std::vector<int32> strides_; member in class:tensorflow::__anon38736::ConvBackpropInputOp 652 std::vector<int32> strides_; member in class:tensorflow::__anon38736::ConvBackpropFilterOp [all...] |
/external/tensorflow/tensorflow/core/kernels/ |
extract_image_patches_op.cc | 56 ParseAttributeVec4(context, "strides", &strides_); 77 const int stride_rows = strides_[1]; 78 const int stride_cols = strides_[2]; 115 std::vector<int32> strides_; member in class:tensorflow::ExtractImagePatchesOp
|
depthwise_conv_op.cc | 263 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 269 OP_REQUIRES(context, strides_.size() == 4, 272 stride_ = GetTensorDim(strides_, data_format_, 'H'); 273 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); 274 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); 275 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); 407 std::vector<int32> strides_; member in class:tensorflow::DepthwiseConv2dNativeOp
|
dilation_ops.cc | 118 ParseAttributes(context, &strides_, &rates_, &padding_); 130 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, 155 std::vector<int32> strides_; member in class:tensorflow::DilationOp 216 ParseAttributes(context, &strides_, &rates_, &padding_); 229 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, 262 std::vector<int32> strides_; member in class:tensorflow::DilationBackpropInputOp 336 ParseAttributes(context, &strides_, &rates_, &padding_); 349 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, 382 std::vector<int32> strides_; member in class:tensorflow::DilationBackpropFilterOp
|
conv_ops_using_gemm.cc | 433 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 441 OP_REQUIRES(context, strides_.size() == 4, 444 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); 445 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); 517 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); 518 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); 556 std::vector<int32> strides_; member in class:tensorflow::Conv2DUsingGemmOp
|
conv_ops.cc | 261 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 272 OP_REQUIRES(context, strides_.size() == 4, 275 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); 276 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); 277 const int64 stride_h = GetTensorDim(strides_, data_format_, 'H'); 278 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); 367 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); 368 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); 429 std::vector<int32> strides_; member in class:tensorflow::Conv2DOp [all...] |
depthwise_conv_grad_op.cc | 527 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 528 OP_REQUIRES(context, strides_.size() == 4, 537 stride_ = GetTensorDim(strides_, data_format_, 'H'); 538 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); 539 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); 540 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); 587 std::vector<int32> strides_; member in class:tensorflow::DepthwiseConv2dNativeBackpropInputOp 965 std::vector<int32> strides_; member in class:tensorflow::DepthwiseConv2dNativeBackpropFilterOp [all...] |
mkl_conv_grad_filter_ops.cc | 70 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 71 int stride_n = GetTensorDim(strides_, data_format_, 'N'); 72 int stride_c = GetTensorDim(strides_, data_format_, 'C'); 134 input_shape, filter_shape, out_backprop_shape, strides_, 422 std::vector<int32> strides_; member in class:tensorflow::MklConv2DCustomBackpropFilterOp
|
mkl_conv_ops.h | 60 std::vector<int32> strides_; member in class:tensorflow::MklDnnConvUtil 67 : context_(context), strides_(strides), padding_(pad), data_format_(fm) {} 76 int stride_rows = GetTensorDim(strides_, data_format_, 'H'); 77 int stride_cols = GetTensorDim(strides_, data_format_, 'W'); 340 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 341 int stride_n = GetTensorDim(strides_, data_format_, 'N'); 342 int stride_c = GetTensorDim(strides_, data_format_, 'C'); 413 MklDnnConvUtil conv_utl(context, strides_, padding_, data_format_); 552 std::vector<int32> strides_; member in class:tensorflow::MklConv2DBackpropCommonOp
|
quantized_conv_ops.cc | 454 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 455 OP_REQUIRES(context, strides_.size() == 4, 458 OP_REQUIRES(context, strides_[1] == strides_[2], 463 context, (strides_[0] == 1 && strides_[3] == 1), 538 const int stride = strides_[1]; 583 std::vector<int32> strides_; member in class:tensorflow::QuantizedConv2DOp
|
conv_grad_filter_ops.cc | 190 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 191 OP_REQUIRES(context, strides_.size() == 4, 195 context, (strides_[0] == 1 && strides_[3] == 1), 198 OP_REQUIRES(context, strides_[1] > 0 && strides_[2] > 0, 235 out_backprop.shape(), strides_, padding_, data_format_, &dims)); 283 std::vector<int32> strides_; member in class:tensorflow::Conv2DFastBackpropFilterOp 303 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 304 OP_REQUIRES(context, strides_.size() == 4 497 std::vector<int32> strides_; member in class:tensorflow::Conv2DCustomBackpropFilterOp 620 std::vector<int32> strides_; member in class:tensorflow::Conv2DSlowBackpropFilterOp [all...] |
conv_grad_input_ops.cc | 193 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 194 OP_REQUIRES(context, strides_.size() == 4, 198 context, (strides_[0] == 1 && strides_[3] == 1), 201 OP_REQUIRES(context, strides_[1] > 0 && strides_[2] > 0, 238 strides_, padding_, data_format_, &dims)); 288 std::vector<int32> strides_; member in class:tensorflow::Conv2DFastBackpropInputOp 308 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 309 OP_REQUIRES(context, strides_.size() == 4 569 std::vector<int32> strides_; member in class:tensorflow::Conv2DCustomBackpropInputOp 689 std::vector<int32> strides_; member in class:tensorflow::Conv2DSlowBackpropInputOp [all...] |
conv_ops_fused.cc | 631 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 632 OP_REQUIRES(context, strides_.size() == 4, 635 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N'); 636 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); 865 std::vector<int32> strides_; member in class:tensorflow::FusedResizeConv2DUsingGemmOp [all...] |
mkl_conv_ops.cc | 71 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 76 OP_REQUIRES(context, strides_.size() == 4, 80 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); 81 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); 171 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); 172 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); 481 std::vector<int32> strides_; member in class:tensorflow::MklConv2DOp 494 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 499 OP_REQUIRES(context, strides_.size() == 4, 503 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N') 658 std::vector<int32> strides_; member in class:tensorflow::MklConv2DOp [all...] |
/external/tensorflow/tensorflow/core/kernels/neon/ |
neon_depthwise_conv_op.cc | 46 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); 47 OP_REQUIRES(context, strides_.size() == 4, 50 OP_REQUIRES(context, strides_[1] == strides_[2], 55 context, (strides_[0] == 1 && strides_[3] == 1), 88 const int32 stride = strides_[1]; 189 std::vector<int32> strides_; member in class:tensorflow::NeonDepthwiseConv2dNativeOp
|
/external/tensorflow/tensorflow/core/util/ |
mkl_util.h | 75 if (strides_) delete[] strides_; 101 strides_ = new size_t[dimension]; 105 strides_[ii] = strides[ii]; 162 const size_t* GetStrides() const { return strides_; } 219 // strides_ 230 // sizeof(strides_) 271 strides_ = new size_t[dimension_]; 276 strides_[i] = reinterpret_cast<const size_t*>( 301 strides_[i] 323 size_t* strides_ = nullptr; \/\/ Required by MKL for conversions member in class:tensorflow::MklShape [all...] |
/external/tensorflow/tensorflow/stream_executor/ |
dnn.h | 653 SetDim(&strides_, DimIndex::Y, value); 657 SetDim(&strides_, DimIndex::X, value); 661 SetDim(&strides_, dim, value); 682 int64 vertical_stride() const { return GetDim(strides_, DimIndex::Y); } 683 int64 horizontal_stride() const { return GetDim(strides_, DimIndex::X); } 684 int64 stride(DimIndex dim) const { return GetDim(strides_, dim); } 687 std::vector<int64> strides() const { return strides_; } 698 std::vector<int64> strides_; member in class:perftools::gputools::dnn::PoolingDescriptor [all...] |