Searched refs:kernel_sizes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 652 std::vector<int32> kernel_sizes; in AvgPoolShape() local
653 TF_RETURN_IF_ERROR(c->GetAttr("ksize", &kernel_sizes)); in AvgPoolShape()
654 if (kernel_sizes.size() != 4) { in AvgPoolShape()
657 kernel_sizes.size()); in AvgPoolShape()
662 int32 kernel_rows = GetTensorDim(kernel_sizes, data_format, 'H'); in AvgPoolShape()
663 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in AvgPoolShape()
802 std::vector<int32> kernel_sizes; in MaxPoolShape() local
803 TF_RETURN_IF_ERROR(c->GetAttr("ksize", &kernel_sizes)); in MaxPoolShape()
804 if (kernel_sizes.size() != 4) { in MaxPoolShape()
807 kernel_sizes.size()); in MaxPoolShape()
[all …]
|
| /external/eigen/bench/tensors/ |
| D | tensor_benchmarks.h | 422 Eigen::array<TensorIndex, 2> kernel_sizes; in convolution() local
423 kernel_sizes[0] = kernel_x; in convolution()
424 kernel_sizes[1] = kernel_y; in convolution()
425 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, kernel_sizes); in convolution()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | atrous_convolution_test.py | 213 kernel_sizes = [1, 3] if padding == "SAME" else range(1, 3)
214 for kernel in kernel_sizes:
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 652 std::vector<int32> kernel_sizes; in RegisterNodeWithPaddingAndStrides() local
653 TF_CHECK_OK(context->GetAttr(KSIZE_ATTR_NAME, &kernel_sizes)); in RegisterNodeWithPaddingAndStrides()
654 const int ksize_id = RegisterConstantShape(kernel_sizes); in RegisterNodeWithPaddingAndStrides()
|