| /external/tensorflow/tensorflow/core/kernels/ |
| eigen_benchmark.h | 45 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) {
49 filter_dims[3]); // filter_count
54 static_cast<Scalar*>(device_.allocate(BufferSize(filter_dims)));
59 device_.memset(filter_data, 123, BufferSize(filter_dims));
62 Filter filter(filter_data, filter_dims);
78 Dimensions filter_dims) {
85 filter_dims[3]); // filter_count
92 static_cast<Scalar*>(device_.allocate(BufferSize(filter_dims)));
98 device_.memset(filter_data, 123, BufferSize(filter_dims));
101 Filter filter(filter_data, filter_dims);
[all...] |
| eigen_benchmark_cpu_test.cc | 46 typename Benchmark::Dimensions filter_dims(filter_height, filter_width,
49 benchmark.SpatialConvolution(input_dims, filter_dims);
75 typename Benchmark::Dimensions filter_dims(filter_height, filter_width,
78 benchmark.SpatialConvolutionBackwardInput(input_dims, filter_dims);
103 typename Benchmark::Dimensions filter_dims(filter_height, filter_width,
106 benchmark.SpatialConvolutionBackwardKernel(input_dims, filter_dims);
108 auto num_computed_elements = filter_dims.TotalSize();
268 typename Benchmark::Dimensions filter_dims(
271 benchmark.CuboidConvolution(input_dims, filter_dims);
298 typename Benchmark::Dimensions filter_dims(
[all...] |
| mkl_conv_ops.h | 188 memory::dims* filter_dims,
190 CHECK_NOTNULL(filter_dims);
235 *filter_dims = mkldnn_sizes;
243 *filter_dims = mkldnn_sizes;
272 *filter_dims = mkldnn_sizes;
282 memory::dims* filter_dims,
284 CHECK_NOTNULL(filter_dims);
286 GetTfShape(context_, filter_index), filter_dims, local
517 memory::dims* input_dims, memory::dims* filter_dims,
523 CHECK_NOTNULL(filter_dims);
[all...] |
| mkl_conv_grad_input_ops.cc | 56 memory::dims filter_dims; member in struct:tensorflow::MklConvBwdInputParams
64 MklConvBwdInputParams(memory::dims diff_src_dims, memory::dims filter_dims,
69 filter_dims(filter_dims),
181 {convBwdInputDims.filter_dims}, MklDnnType<T>(), memory::format::any));
277 key_creator.AddAsKey(convBwdInputDims.filter_dims);
|
| mkl_pooling_ops_common.h | 48 memory::dims filter_dims; member in struct:tensorflow::MklPoolingParams
56 memory::dims filter_dims, memory::dims strides,
61 filter_dims(filter_dims),
190 key_creator.AddAsKey(fwdParams.filter_dims);
342 key_creator.AddAsKey(bwdParams.filter_dims);
522 memory::dims* filter_dims, memory::dims* strides,
527 *filter_dims =
537 *filter_dims =
|
| mkl_avgpooling_op.cc | 477 memory::dims filter_dims, strides, padding_left, padding_right; variable
479 this->PoolParamsToDims(&pool_params, &filter_dims, &strides,
504 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims,
599 memory::dims filter_dims, strides, padding_left, padding_right; variable
600 this->PoolParamsToDims(&pool_params, &filter_dims, &strides,
624 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims,
|
| mkl_maxpooling_op.cc | 568 memory::dims filter_dims, strides, padding_left, padding_right; variable
569 this->PoolParamsToDims(&pool_params, &filter_dims, &strides,
581 MklPoolingParams fwdParams(src_dims, output_dims_mkl_order, filter_dims,
710 memory::dims filter_dims, strides, padding_left, padding_right; variable
711 this->PoolParamsToDims(&pool_params, &filter_dims, &strides,
734 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims,
[all...] |
| mkl_conv_ops.cc | 68 memory::dims filter_dims; member in struct:tensorflow::MklConvFwdParams
82 MklConvFwdParams(memory::dims src_dims, memory::dims filter_dims,
87 filter_dims(filter_dims),
220 {convFwdDims.filter_dims}, MklDnnType<Tfilter>(), memory::format::any));
366 key_creator.AddAsKey(convFwdDims.filter_dims);
562 mkl_context.filter_dims = filter.dims();
655 mkl_filter_output_mkl_shape.SetTfDimOrder(mkl_context.filter_dims,
691 int filter_dims; member in struct:tensorflow::MklConvOp::__anon45425
714 CHECK_EQ(dnnLayoutCreate_F32(<_filter, filter_dims, filter_sizes
940 memory::dims src_dims, filter_dims, padding_left, padding_right, variable
[all...] |
| mkl_pooling_ops_common.cc | 65 *context_.dst_md, fwdParams.strides, fwdParams.filter_dims,
152 bwdParams.strides, bwdParams.filter_dims, bwdParams.padding_left,
159 *context_.diff_dst_md, bwdParams.strides, bwdParams.filter_dims,
|
| eigen_spatial_convolutions_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| convolution_test.cc | 414 std::vector<int64> filter_dims = {2, 2, 2, 3, 3}; local
416 Shape filter_shape = ShapeUtil::MakeShape(F32, filter_dims);
450 auto filter_r5 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie();
480 std::vector<int64> filter_dims = {3, 3, 5, 3}; local
482 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims);
513 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie();
539 std::vector<int64> filter_dims = {3, 3, 1, 15}; local
541 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims);
573 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie();
605 std::vector<int64> filter_dims = {3, 3, 1, 5} local
675 std::vector<int64> filter_dims = {3, 3, 1, 512}; local
739 std::vector<int64> filter_dims = {3, 3, 1, 512}; local
808 std::vector<int64> filter_dims = {3, 3, 1, 512}; local
877 std::vector<int64> filter_dims = {3, 3, 1, 512}; local
948 std::vector<int64> filter_dims = {3, 3, 1, 5}; local
1023 std::vector<int64> filter_dims = {3, 3, 1, 160}; local
1087 std::vector<int64> filter_dims = {3, 3, 1, 160}; local
1157 std::vector<int64> filter_dims = {3, 3, 1, 160}; local
1227 std::vector<int64> filter_dims = {3, 3, 1, 1024}; local
1290 std::vector<int64> filter_dims = {2, 2, 2, 12}; local
1355 std::vector<int64> filter_dims = {2, 2, 128, 512}; local
1419 std::vector<int64> filter_dims = {2, 2, 128, 8}; local
1483 std::vector<int64> filter_dims = {2, 2, 3, 4}; local
1547 std::vector<int64> filter_dims = {2, 2, 4, 3}; local
1613 std::vector<int64> filter_dims = {1, 1, 3, 4}; local
1739 std::vector<int64> filter_dims = {window_size, input_feature, local
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| legacy_optimized_ops.h | 139 const float* filter_data, const Dims<4>& filter_dims,
161 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims),
166 const float* filter_data, const Dims<4>& filter_dims,
173 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data,
182 const float* filter_data, const Dims<4>& filter_dims,
189 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data,
198 const float* filter_data, const Dims<4>& filter_dims,
202 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data,
209 const Dims<4>& filter_dims, int32 filter_offset,
238 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims)
[all...] |
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| depthwise_conv_test.cc | 38 TfLiteIntArray* filter_dims = IntArrayFromInitializer(filter_dims_data); local
48 CreateFloatTensor(filter_data, filter_dims, "filter_tensor"),
67 int output_depth = filter_dims->data[3];
119 TfLiteIntArray* filter_dims = IntArrayFromInitializer(filter_dims_data); local
130 CreateQuantizedTensor(filter_data, filter_dims, "filter_tensor",
147 int output_depth = filter_dims->data[3];
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| legacy_reference_ops.h | 40 const float* filter_data, const Dims<4>& filter_dims,
62 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims),
67 const float* filter_data, const Dims<4>& filter_dims,
74 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data,
83 const float* filter_data, const Dims<4>& filter_dims,
90 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data,
99 const float* filter_data, const Dims<4>& filter_dims,
103 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data,
110 const Dims<4>& filter_dims, int32 filter_offset,
139 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims)
[all...] |
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| depthwiseconv_float.h | 554 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/ |
| optimized_ops.h | [all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | [all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 1178 int filter_dims[] = {static_cast<int>(params_size_in_bytes), 1, 1}; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator_test.cc | 1367 std::vector<int64> filter_dims = {2, 2, 2, 8}; local
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| convert_nodes_test.cc | 3239 std::vector<int> filter_dims; member in struct:tensorflow::tensorrt::convert::TestParams
[all...] |
