Searched refs:channels_in (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | decode_image_op_test.py | 103 for channels_in, filename in inputs:
111 self.assertEqual(image0.shape, (26, 51, channels or channels_in))
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | conv.cc | 270 int channels_in = filter->dims->data[3]; in Prepare() local
406 const int height = NumElements(input) / channels_in; in Prepare()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 789 channels_in = input_shape[0]
805 np.prod(kernel_sizes) * channels_in,
811 output_shape + (channels_in, np.prod(kernel_sizes), filters)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 3889 for channels_in, filename in inputs:
3895 self.assertEqual(image0.shape, (26, 51, channels or channels_in))
3896 if channels == channels_in:
|