| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils.py | 210 def conv_kernel_mask(input_shape, kernel_shape, strides, padding): argument
289 def conv_kernel_idxs(input_shape, kernel_shape, strides, padding, filters_in, argument
383 def conv_connected_inputs(input_shape, kernel_shape, output_position, strides, argument
442 def conv_output_shape(input_shape, kernel_shape, strides, padding): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_support_utils.cc | 32 const Shape& kernel_shape = conv.operand(1)->shape(); in CudnnSupportsOptimizedIntegerConvolution() local
|
| D | cudnn_pad_for_convolutions.cc | 300 const Shape& kernel_shape = conv->operand(1)->shape(); in TryResolvePaddedShapesForIntegerConvolution() local
|
| D | cudnn_vectorize_convolutions.cc | 264 const Shape& kernel_shape = conv->operand(1)->shape(); in TryRevectorizeConv() local
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuWinogradConv2d.cpp | 93 Tensor4DShape kernel_shape{ internal_get_shape(weights) }; in get_winograd_kernel_implementation() local
172 const Tensor4DShape kernel_shape{ internal_get_shape(weights) }; in configure() local
303 const Tensor4DShape kernel_shape{ internal_get_shape(weights) }; in validate() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 53 const Shape& kernel_shape = convolution.operand(1)->shape(); in PotentiallyImplementedAsEigenConvolution() local
|
| D | ir_emitter.cc | 956 const Shape& kernel_shape = convolution->operand(1)->shape(); in HandleConvolution() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 109 std::vector<int64_t> kernel_shape(num_dims, 1); in Compile() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | morphological_ops_test.py | 183 def _ConstructAndTestGradient(self, image_shape, kernel_shape, strides, rates, argument
489 def _ConstructAndTestGradient(self, image_shape, kernel_shape, strides, rates, argument
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 122 const int32_t* kernel_shape, in GetConvOutputSizeVALID()
219 const int32_t* __restrict kernel_shape, in DepthwiseConv2dNative()
310 const int32_t* __restrict kernel_shape, in DepthwiseConv2dNativeVALID()
325 const int32_t* __restrict kernel_shape, in DepthwiseConv2dNativeSAME()
573 const int32_t* __restrict kernel_shape, in Im2Row()
691 const int32_t* __restrict kernel_shape, in Im2RowVALID()
704 const int32_t* __restrict kernel_shape, in Im2RowSAME()
|
| D | onedevice_train_palm_detection_filter_inference.cc | 123 const int32_t* kernel_shape, in GetConvOutputSizeVALID()
220 const int32_t* __restrict kernel_shape, in DepthwiseConv2dNative()
311 const int32_t* __restrict kernel_shape, in DepthwiseConv2dNativeVALID()
326 const int32_t* __restrict kernel_shape, in DepthwiseConv2dNativeSAME()
574 const int32_t* __restrict kernel_shape, in Im2Row()
692 const int32_t* __restrict kernel_shape, in Im2RowVALID()
705 const int32_t* __restrict kernel_shape, in Im2RowSAME()
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 122 const int32_t* kernel_shape, in GetConvOutputSizeVALID()
219 const int32_t* __restrict kernel_shape, in DepthwiseConv2dNative()
310 const int32_t* __restrict kernel_shape, in DepthwiseConv2dNativeVALID()
325 const int32_t* __restrict kernel_shape, in DepthwiseConv2dNativeSAME()
573 const int32_t* __restrict kernel_shape, in Im2Row()
691 const int32_t* __restrict kernel_shape, in Im2RowVALID()
704 const int32_t* __restrict kernel_shape, in Im2RowSAME()
|
| /external/ComputeLibrary/src/core/NEON/kernels/assembly/ |
| D | winograd.hpp | 45 Shape2D kernel_shape; member
|
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.cpp | 1318 …std::vector<uint32_t> kernel_shape = ReadMandatoryNodeUint32ListAttribute(node, "kernel_shape"); /… in AddPoolingLayer() local
1802 … std::vector<uint32_t> kernel_shape = ReadOptionalNodeUint32ListAttribute(node, "kernel_shape"); in ParseConv() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2360 Tensor kernel_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
2438 Tensor kernel_shape = test::AsTensor<int32>( in TEST_F() local
2560 Tensor kernel_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 3547 const auto& kernel_shape = kernel->shape(); in GetConvolutionDetails() local
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 1511 auto kernel_shape = xla::ShapeUtil::MakeShape(xla::BF16, {3, 3, 5, 5}); in TEST() local
|