| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils.py | 214 def conv_kernel_mask(input_shape, kernel_shape, strides, padding): argument
293 def conv_kernel_idxs(input_shape, kernel_shape, strides, padding, filters_in, argument
387 def conv_connected_inputs(input_shape, kernel_shape, output_position, strides, argument
446 def conv_output_shape(input_shape, kernel_shape, strides, padding): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 54 const Shape& kernel_shape = convolution.operand(1)->shape(); in PotentiallyImplementedAsEigenConvolution() local
|
| D | ir_emitter.cc | 913 const Shape& kernel_shape = convolution->operand(1)->shape(); in HandleConvolution() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 635 def get_locallyconnected_mask(input_shape, kernel_shape, strides, padding, argument
744 def local_conv_sparse_matmul(inputs, kernel, kernel_idxs, kernel_shape, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 109 std::vector<int64> kernel_shape(num_dims, 1); in Compile() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | morphological_ops_test.py | 185 def _ConstructAndTestGradient(self, image_shape, kernel_shape, strides, rates, argument
461 def _ConstructAndTestGradient(self, image_shape, kernel_shape, strides, rates, argument
|
| /external/XNNPACK/bench/ |
| D | convolution.cc | 875 arm_compute::TensorShape kernel_shape( in armcl_convolution_f32() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1632 Tensor kernel_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
1706 Tensor kernel_shape = test::AsTensor<int32>( in TEST_F() local
1811 Tensor kernel_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 2596 const auto& kernel_shape = kernel->shape(); in GetConvolutionDetails() local
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 1504 auto kernel_shape = xla::ShapeUtil::MakeShape(xla::BF16, {3, 3, 5, 5}); in TEST() local
|