| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_benchmark.h | 45 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution()
78 Dimensions filter_dims) { in SpatialConvolutionBackwardInput()
119 Dimensions filter_dims) { in SpatialConvolutionBackwardKernel()
180 void CuboidConvolution(Dimensions input_dims, Dimensions filter_dims) { in CuboidConvolution()
214 Dimensions filter_dims) { in CuboidConvolutionBackwardInput()
257 Dimensions filter_dims) { in CuboidConvolutionBackwardKernel()
|
| D | eigen_benchmark_cpu_test.cc | 46 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolution() local
75 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardInput() local
103 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardKernel() local
268 typename Benchmark::Dimensions filter_dims( in CuboidConvolution() local
298 typename Benchmark::Dimensions filter_dims( in CuboidConvolutionBackwardInput() local
327 typename Benchmark::Dimensions filter_dims( in CuboidConvolutionBackwardKernel() local
|
| D | mkl_conv_ops.h | 188 memory::dims* filter_dims, in GetFilterSizeInMklOrder()
282 memory::dims* filter_dims, in GetFilterSizeInMklOrder()
|
| D | mkl_avgpooling_op.cc | 477 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
599 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
|
| D | mkl_maxpooling_op.cc | 568 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
710 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
|
| D | mkl_pooling_ops_common.h | 48 memory::dims filter_dims; member
522 memory::dims* filter_dims, memory::dims* strides, in PoolParamsToDims()
|
| D | mkl_conv_ops.cc | 68 memory::dims filter_dims; member
691 int filter_dims; member
940 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
|
| D | mkl_conv_grad_input_ops.cc | 56 memory::dims filter_dims; member
|
| D | eigen_spatial_convolutions_test.cc | 1572 Dimensions filter_dims(filter_count, filter_rows, filter_cols, input_depth); in PackLhsHelper() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 414 std::vector<int64> filter_dims = {2, 2, 2, 3, 3}; in XLA_TEST_F() local
480 std::vector<int64> filter_dims = {3, 3, 5, 3}; in RunTest() local
539 std::vector<int64> filter_dims = {3, 3, 1, 15}; in RunTest() local
605 std::vector<int64> filter_dims = {3, 3, 1, 5}; in RunTest() local
675 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
739 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
808 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
877 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
948 std::vector<int64> filter_dims = {3, 3, 1, 5}; in RunTest() local
1023 std::vector<int64> filter_dims = {3, 3, 1, 160}; in RunTest() local
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 139 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
166 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
182 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
198 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
209 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
244 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
264 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
287 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
358 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
383 const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 40 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
67 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
83 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
99 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
110 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
145 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
165 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
188 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
204 const float* filter_data, const Dims<4>& filter_dims, in Conv()
230 const float* filter_data, const Dims<4>& filter_dims, in Conv()
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | depthwise_conv_test.cc | 38 TfLiteIntArray* filter_dims = IntArrayFromInitializer(filter_dims_data); in TestDepthwiseConvFloat() local
119 TfLiteIntArray* filter_dims = IntArrayFromInitializer(filter_dims_data); in TestDepthwiseConvQuantized() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1210 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
1324 const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
1350 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
2026 const Dims<4>& filter_dims, int stride_width, in DilatedIm2col()
2177 const float* filter_data, const Dims<4>& filter_dims, in Conv()
2287 const int8_t* filter_data, const Dims<4>& filter_dims, in HybridConv()
2314 const float* filter_data, const Dims<4>& filter_dims, in Conv()
2333 const float* filter_data, const Dims<4>& filter_dims, in Conv()
2350 const float* filter_data, const Dims<4>& filter_dims, in Conv()
2455 const Dims<4>& filter_dims, int32 filter_offset, in Conv()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 2370 inline bool IsConv1x1StrideNot1(memory::dims filter_dims, in IsConv1x1StrideNot1()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3239 std::vector<int> filter_dims; in TEST_F() member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 1367 std::vector<int64> filter_dims = {2, 2, 2, 8}; in TEST_P() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1178 int filter_dims[] = {static_cast<int>(params_size_in_bytes), 1, 1}; in Create() local
|