Home
last modified time | relevance | path

Searched refs:filter_dims (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h45 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution() argument
49 filter_dims[3]); // filter_count in SpatialConvolution()
54 static_cast<Scalar*>(device_.allocate(BufferSize(filter_dims))); in SpatialConvolution()
59 device_.memset(filter_data, 123, BufferSize(filter_dims)); in SpatialConvolution()
62 Filter filter(filter_data, filter_dims); in SpatialConvolution()
78 Dimensions filter_dims) { in SpatialConvolutionBackwardInput() argument
85 filter_dims[3]); // filter_count in SpatialConvolutionBackwardInput()
92 static_cast<Scalar*>(device_.allocate(BufferSize(filter_dims))); in SpatialConvolutionBackwardInput()
98 device_.memset(filter_data, 123, BufferSize(filter_dims)); in SpatialConvolutionBackwardInput()
101 Filter filter(filter_data, filter_dims); in SpatialConvolutionBackwardInput()
[all …]
Deigen_benchmark_cpu_test.cc46 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolution() local
49 benchmark.SpatialConvolution(input_dims, filter_dims); in SpatialConvolution()
75 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardInput() local
78 benchmark.SpatialConvolutionBackwardInput(input_dims, filter_dims); in SpatialConvolutionBackwardInput()
103 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardKernel() local
106 benchmark.SpatialConvolutionBackwardKernel(input_dims, filter_dims); in SpatialConvolutionBackwardKernel()
108 auto num_computed_elements = filter_dims.TotalSize(); in SpatialConvolutionBackwardKernel()
268 typename Benchmark::Dimensions filter_dims( in CuboidConvolution() local
271 benchmark.CuboidConvolution(input_dims, filter_dims); in CuboidConvolution()
298 typename Benchmark::Dimensions filter_dims( in CuboidConvolutionBackwardInput() local
[all …]
Dmkl_conv_ops.h188 memory::dims* filter_dims, in GetFilterSizeInMklOrder() argument
190 CHECK_NOTNULL(filter_dims); in GetFilterSizeInMklOrder()
235 *filter_dims = mkldnn_sizes; in GetFilterSizeInMklOrder()
243 *filter_dims = mkldnn_sizes; in GetFilterSizeInMklOrder()
272 *filter_dims = mkldnn_sizes; in GetFilterSizeInMklOrder()
282 memory::dims* filter_dims, in GetFilterSizeInMklOrder() argument
284 CHECK_NOTNULL(filter_dims); in GetFilterSizeInMklOrder()
286 GetTfShape(context_, filter_index), filter_dims, in GetFilterSizeInMklOrder()
517 memory::dims* input_dims, memory::dims* filter_dims,
523 CHECK_NOTNULL(filter_dims);
[all …]
Dmkl_pooling_ops_common.h48 memory::dims filter_dims; member
56 memory::dims filter_dims, memory::dims strides, in MklPoolingParams()
61 filter_dims(filter_dims), in MklPoolingParams()
190 key_creator.AddAsKey(fwdParams.filter_dims); in CreateKey()
342 key_creator.AddAsKey(bwdParams.filter_dims); in CreateKey()
522 memory::dims* filter_dims, memory::dims* strides, in PoolParamsToDims() argument
527 *filter_dims = in PoolParamsToDims()
537 *filter_dims = in PoolParamsToDims()
Dmkl_conv_grad_input_ops.cc56 memory::dims filter_dims; member
64 MklConvBwdInputParams(memory::dims diff_src_dims, memory::dims filter_dims, in MklConvBwdInputParams()
69 filter_dims(filter_dims), in MklConvBwdInputParams()
181 {convBwdInputDims.filter_dims}, MklDnnType<T>(), memory::format::any)); in Setup()
277 key_creator.AddAsKey(convBwdInputDims.filter_dims); in CreateKey()
Dmkl_avgpooling_op.cc477 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
479 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
504 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims, in Compute()
599 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
600 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
624 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims, in Compute()
Dmkl_conv_ops.cc68 memory::dims filter_dims; member
82 MklConvFwdParams(memory::dims src_dims, memory::dims filter_dims, in MklConvFwdParams()
87 filter_dims(filter_dims), in MklConvFwdParams()
220 {convFwdDims.filter_dims}, MklDnnType<Tfilter>(), memory::format::any)); in Setup()
366 key_creator.AddAsKey(convFwdDims.filter_dims); in CreateKey()
562 mkl_context.filter_dims = filter.dims(); in Compute()
655 mkl_filter_output_mkl_shape.SetTfDimOrder(mkl_context.filter_dims, in Compute()
691 int filter_dims; member
714 CHECK_EQ(dnnLayoutCreate_F32(&lt_filter, filter_dims, filter_sizes, in MklCreateInputLayouts()
940 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
[all …]
Dmkl_maxpooling_op.cc568 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
569 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
581 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims, in Compute()
710 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
711 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
734 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims, in Compute()
Dmkl_pooling_ops_common.cc65 *context_.dst_md, fwdParams.strides, fwdParams.filter_dims, in Setup()
152 bwdParams.strides, bwdParams.filter_dims, bwdParams.padding_left, in Setup()
159 *context_.diff_dst_md, bwdParams.strides, bwdParams.filter_dims, in Setup()
Deigen_spatial_convolutions_test.cc1572 Dimensions filter_dims(filter_count, filter_rows, filter_cols, input_depth); in PackLhsHelper() local
1642 Tensor<float, 4> packed(filter_dims); in PackLhsHelper()
1646 size_t input_bytes = filter_dims.TotalSize() * sizeof(float); in PackLhsHelper()
1656 filters.emplace_back(filter_dims); in PackLhsHelper()
1659 ArgType tensor_map(filters[i].data(), filter_dims); in PackLhsHelper()
1672 const Index packed_total_size = filter_dims.TotalSize(); in PackLhsHelper()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc414 std::vector<int64> filter_dims = {2, 2, 2, 3, 3}; in XLA_TEST_F() local
416 Shape filter_shape = ShapeUtil::MakeShape(F32, filter_dims); in XLA_TEST_F()
450 auto filter_r5 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in XLA_TEST_F()
480 std::vector<int64> filter_dims = {3, 3, 5, 3}; in RunTest() local
482 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest()
513 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest()
539 std::vector<int64> filter_dims = {3, 3, 1, 15}; in RunTest() local
541 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest()
573 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest()
605 std::vector<int64> filter_dims = {3, 3, 1, 5}; in RunTest() local
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h139 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
161 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
166 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
173 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
182 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
189 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
198 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
202 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
209 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv() argument
238 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h40 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
62 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
67 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
74 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
83 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
90 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
99 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
103 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
110 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv() argument
139 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv_test.cc38 TfLiteIntArray* filter_dims = IntArrayFromInitializer(filter_dims_data); in TestDepthwiseConvFloat() local
48 CreateFloatTensor(filter_data, filter_dims, "filter_tensor"), in TestDepthwiseConvFloat()
67 int output_depth = filter_dims->data[3]; in TestDepthwiseConvFloat()
119 TfLiteIntArray* filter_dims = IntArrayFromInitializer(filter_dims_data); in TestDepthwiseConvQuantized() local
130 CreateQuantizedTensor(filter_data, filter_dims, "filter_tensor", in TestDepthwiseConvQuantized()
147 int output_depth = filter_dims->data[3]; in TestDepthwiseConvQuantized()
/external/tensorflow/tensorflow/core/kernels/neon/
Ddepthwiseconv_float.h554 const float* filter_data, const Dims<4>& filter_dims,
565 const int output_depth = MatchingArraySize(filter_dims, 0, output_dims, 0);
569 const int filter_height = ArraySize(filter_dims, 2);
570 const int filter_width = ArraySize(filter_dims, 1);
655 filter_data + filter_y * filter_dims.strides[2],
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h1210 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected() argument
1228 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in FullyConnected()
1324 const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected() argument
1340 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in FullyConnected()
1350 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected() argument
1362 FullyConnected(input_data, input_dims, input_offset, filter_data, filter_dims, in FullyConnected()
2026 const Dims<4>& filter_dims, int stride_width, in DilatedIm2col() argument
2042 DimsToShape(filter_dims), DimsToShape(output_dims), in DilatedIm2col()
2177 const float* filter_data, const Dims<4>& filter_dims, in Conv() argument
2196 Conv(op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in Conv()
[all …]
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h2370 inline bool IsConv1x1StrideNot1(memory::dims filter_dims, in IsConv1x1StrideNot1() argument
2372 if (filter_dims.size() != 4 || strides.size() != 2) return false; in IsConv1x1StrideNot1()
2374 return ((filter_dims[2] == 1) && (filter_dims[3] == 1) && in IsConv1x1StrideNot1()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc1367 std::vector<int64> filter_dims = {2, 2, 2, 8}; in TEST_P() local
1369 Shape filter_shape = ShapeUtil::MakeShapeWithType<float>(filter_dims); in TEST_P()
1406 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in TEST_P()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1178 int filter_dims[] = {static_cast<int>(params_size_in_bytes), 1, 1}; in Create() local
1182 /*nbDims=*/sizeof(filter_dims) / sizeof(filter_dims[0]), in Create()
1183 /*filterDimA=*/filter_dims)); in Create()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc3239 std::vector<int> filter_dims; in TEST_F() member
3345 AddTestWeights<float>("weights", ok_params[i].filter_dims, in TEST_F()