Searched refs:kernel_sizes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 745 std::vector<int32> kernel_sizes; in AvgPoolShape() local
746 TF_RETURN_IF_ERROR(c->GetAttr("ksize", &kernel_sizes)); in AvgPoolShape()
747 if (kernel_sizes.size() != 4) { in AvgPoolShape()
750 kernel_sizes.size()); in AvgPoolShape()
755 int32 kernel_rows = GetTensorDim(kernel_sizes, data_format, 'H'); in AvgPoolShape()
756 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in AvgPoolShape()
895 std::vector<int32> kernel_sizes; in MaxPoolShape() local
896 TF_RETURN_IF_ERROR(c->GetAttr("ksize", &kernel_sizes)); in MaxPoolShape()
897 if (kernel_sizes.size() != 4) { in MaxPoolShape()
900 kernel_sizes.size()); in MaxPoolShape()
[all …]
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | conv2d_test.py | 72 kernel_sizes, argument
81 for kernel_size in kernel_sizes:
102 kernel_sizes=[(3, 3), (3, 2)],
127 kernel_sizes=[(3, 3), (3, 2)],
|
| /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 | 215 kernel_sizes = [1, 3] if padding == "SAME" else range(1, 3)
216 for kernel in kernel_sizes:
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 798 kernel_sizes = (kernel_size,) * dim
805 np.prod(kernel_sizes) * channels_in,
811 output_shape + (channels_in, np.prod(kernel_sizes), filters)
819 kernel_sizes,
836 kernel_sizes,
856 kernel_sizes, argument
866 np.prod(kernel_sizes) * input_shape[-1],
872 kernel_sizes,
879 kernel_sizes,
885 kernel_sizes,
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 659 std::vector<int32> kernel_sizes; in RegisterNodeWithPaddingAndStrides() local
660 TF_CHECK_OK(context->GetAttr(KSIZE_ATTR_NAME, &kernel_sizes)); in RegisterNodeWithPaddingAndStrides()
661 const int ksize_id = RegisterConstantShape(kernel_sizes); in RegisterNodeWithPaddingAndStrides()
|