| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_benchmark.h | 46 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()
|
| D | eigen_benchmark_cpu_test.cc | 44 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
|
| D | conv_grad_filter_ops_benchmark_test.cc | 76 Node* filter_dims = in Conv2DBackpropFilter() local
|
| D | eigen_spatial_convolutions_test.cc | 1595 Dimensions filter_dims(filter_count, filter_rows, filter_cols, input_depth); in PackLhsHelper() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 261 std::vector<int64_t> filter_dims = {2, 2, 2, 3, 3}; in XLA_TEST_F() local
326 std::vector<int64_t> filter_dims = {3, 3, 5, 3}; in RunTest() local
383 std::vector<int64_t> filter_dims = {3, 3, 1, 15}; in RunTest() local
447 std::vector<int64_t> filter_dims = {3, 3, 1, 5}; in RunTest() local
515 std::vector<int64_t> filter_dims = {3, 3, 1, 512}; in RunTest() local
577 std::vector<int64_t> filter_dims = {3, 3, 1, 512}; in RunTest() local
644 std::vector<int64_t> filter_dims = {3, 3, 1, 512}; in RunTest() local
710 std::vector<int64_t> filter_dims = {3, 3, 1, 512}; in RunTest() local
778 std::vector<int64_t> filter_dims = {3, 3, 1, 5}; in RunTest() local
851 std::vector<int64_t> filter_dims = {3, 3, 1, 160}; in RunTest() local
[all …]
|
| D | convolution_test_1d.cc | 82 std::vector<int64_t> filter_dims = {window_size, input_feature, in TestImpl() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_avgpooling_op.cc | 82 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
214 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
|
| D | mkl_maxpooling_op.cc | 121 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
266 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
|
| D | mkl_conv_ops.h | 190 memory::dims* filter_dims, in GetFilterSizeInMklOrder()
292 memory::dims* filter_dims, in GetFilterSizeInMklOrder()
|
| D | mkl_pooling_ops_common.h | 45 memory::dims filter_dims; member
527 memory::dims* filter_dims, memory::dims* strides, in PoolParamsToDims()
|
| D | mkl_conv_grad_input_ops.cc | 49 memory::dims filter_dims; member
|
| D | mkl_conv_ops.cc | 44 memory::dims filter_dims; member
648 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 45 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
72 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
88 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
104 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
115 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
150 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
170 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
193 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
209 const float* filter_data, const Dims<4>& filter_dims, in Conv()
235 const float* filter_data, const Dims<4>& filter_dims, in Conv()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 140 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
172 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
188 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
204 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
283 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
323 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
343 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
366 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
1724 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
1749 const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
[all …]
|
| /external/XNNPACK/test/ |
| D | convolution-2d.cc | 99 std::array<size_t, 4> filter_dims; member in xnnpack::ConvolutionTestBase
|
| D | depthwise-convolution-2d.cc | 103 std::array<size_t, 4> filter_dims; member in xnnpack::DepthwiseConvolutionTestBase
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1557 int filter_dims[] = {static_cast<int>(filter_dim0), 1, 1}; in Create() local
3548 std::vector<int64_t> filter_dims = filter_descriptor.vectorized_dims( in GetCudnnOperationGraph() local
3684 std::vector<int64_t> filter_dims = filter_descriptor.vectorized_dims( in GetCudnnFusedOperationGraph() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 6154 std::vector<int> filter_dims; in TEST_P() member
6280 std::vector<int> filter_dims; in TEST_P() member
6494 std::vector<int> filter_dims; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 1482 std::vector<int64_t> filter_dims = {2, 2, 2, 8}; in TEST_P() local
|