Searched defs:kernel_shape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils.py | 230 def conv_kernel_mask(input_shape, kernel_shape, strides, padding): argument
310 kernel_shape, argument
368 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 | 1136 const Shape& kernel_shape = convolution->operand(1)->shape(); in HandleConvolution() 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/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 585 kernel_shape, argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1631 Tensor kernel_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
1705 Tensor kernel_shape = test::AsTensor<int32>( in TEST_F() local
1810 Tensor kernel_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 800 auto kernel_shape = xla::ShapeUtil::MakeShape(xla::BF16, {3, 3, 5, 5}); in TEST() local
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 2075 kernel_shape, argument
|