Home
last modified time | relevance | path

Searched defs:filter_dims (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h46 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution()
77 Dimensions filter_dims) { in SpatialConvolutionBackwardInput()
116 Dimensions filter_dims) { in SpatialConvolutionBackwardKernel()
177 void CuboidConvolution(Dimensions input_dims, Dimensions filter_dims) { in CuboidConvolution()
209 Dimensions filter_dims) { in CuboidConvolutionBackwardInput()
250 Dimensions filter_dims) { in CuboidConvolutionBackwardKernel()
Deigen_benchmark_cpu_test.cc44 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolution() local
72 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardInput() local
99 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardKernel() local
262 typename Benchmark::Dimensions filter_dims( in CuboidConvolution() local
291 typename Benchmark::Dimensions filter_dims( in CuboidConvolutionBackwardInput() local
319 typename Benchmark::Dimensions filter_dims( in CuboidConvolutionBackwardKernel() local
Dconv_grad_filter_ops_benchmark_test.cc76 Node* filter_dims = in Conv2DBackpropFilter() local
Deigen_spatial_convolutions_test.cc1595 Dimensions filter_dims(filter_count, filter_rows, filter_cols, input_depth); in PackLhsHelper() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc266 std::vector<int64> filter_dims = {2, 2, 2, 3, 3}; in XLA_TEST_F() local
332 std::vector<int64> filter_dims = {3, 3, 5, 3}; in RunTest() local
391 std::vector<int64> filter_dims = {3, 3, 1, 15}; in RunTest() local
457 std::vector<int64> filter_dims = {3, 3, 1, 5}; in RunTest() local
527 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
591 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
660 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
729 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
800 std::vector<int64> filter_dims = {3, 3, 1, 5}; in RunTest() local
875 std::vector<int64> filter_dims = {3, 3, 1, 160}; in RunTest() local
[all …]
Dconvolution_test_1d.cc84 std::vector<int64> filter_dims = {window_size, input_feature, in TestImpl() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc84 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
205 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
Dmkl_maxpooling_op.cc126 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
261 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
Dmkl_conv_ops.h194 memory::dims* filter_dims, in GetFilterSizeInMklOrder()
300 memory::dims* filter_dims, in GetFilterSizeInMklOrder()
Dmkl_pooling_ops_common.h43 memory::dims filter_dims; member
512 memory::dims* filter_dims, memory::dims* strides, in PoolParamsToDims()
Dmkl_conv_grad_input_ops.cc62 memory::dims filter_dims; member
Dmkl_conv_ops.cc60 memory::dims filter_dims; member
627 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h43 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
70 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
86 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
102 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
113 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
148 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
168 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
191 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
207 const float* filter_data, const Dims<4>& filter_dims, in Conv()
233 const float* filter_data, const Dims<4>& filter_dims, in Conv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h138 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
170 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
186 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
202 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
281 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
321 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
341 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
364 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
1722 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
1747 const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
[all …]
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h2032 inline bool IsConv1x1StrideNot1(memory::dims filter_dims, in IsConv1x1StrideNot1()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1484 int filter_dims[] = {static_cast<int>(filter_dim0), 1, 1}; in Create() local
3434 std::vector<int64> filter_dims = filter_descriptor.vectorized_dims( in GetCudnnOperationGraph() local
3576 std::vector<int64> filter_dims = filter_descriptor.vectorized_dims( in GetCudnnFusedOperationGraph() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc4840 std::vector<int> filter_dims; in TEST_P() member
4966 std::vector<int> filter_dims; in TEST_P() member
5140 std::vector<int> filter_dims; member
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc2208 auto filter_dims = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc1444 std::vector<int64> filter_dims = {2, 2, 2, 8}; in TEST_P() local