Searched refs:num_col (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional_recurrent_test.py | 40 num_col = 3
63 'kernel_size': (num_row, num_col),
82 'kernel_size': (num_row, num_col),
89 num_col = 3
105 'kernel_size': (num_row, num_col),
141 num_col = 3
155 'kernel_size': (num_row, num_col),
|
| D | local_test.py | 131 num_col = 10
158 input_shape=(num_samples, num_row, num_col, stack_size))
167 num_col = 10
187 input_shape=(num_samples, num_row, num_col, stack_size))
194 num_col = 7
214 layer.build((num_samples, num_row, num_col, stack_size))
218 np.ones((num_samples, num_row, num_col, stack_size))))
231 layer.build((num_samples, num_row, num_col, stack_size))
|
| D | convolutional_transpose_test.py | 37 num_col = 6
43 input_shape=(num_samples, num_row, num_col, stack_size))
127 num_col = 6
134 input_shape=(num_samples, depth, num_row, num_col, stack_size))
|
| D | separable_convolutional_test.py | 105 num_col = 6
111 input_shape=(num_samples, num_row, num_col, stack_size))
|
| D | convolutional_test.py | 103 num_col = 6
109 input_shape=(num_samples, num_row, num_col, stack_size))
170 num_col = 6
177 input_shape=(num_samples, depth, num_row, num_col, stack_size))
723 num_col = 6
729 input_shape=(num_samples, num_row, num_col, stack_size))
|
| /external/libaom/libaom/av1/encoder/x86/ |
| D | highbd_fwd_txfm_sse4.c | 63 const int num_col) { in fdct4x4_sse4_1() argument
73 int endidx = 3 * num_col; in fdct4x4_sse4_1()
76 endidx -= num_col; in fdct4x4_sse4_1()
77 s1 = _mm_add_epi32(in[num_col], in[endidx]); in fdct4x4_sse4_1()
78 s2 = _mm_sub_epi32(in[num_col], in[endidx]); in fdct4x4_sse4_1()
129 const int num_col) { in fadst4x4_sse4_1() argument
142 int idx = 0 * num_col; in fadst4x4_sse4_1()
145 t = _mm_add_epi32(in[idx], in[idx + num_col]); in fadst4x4_sse4_1()
146 idx += num_col; in fadst4x4_sse4_1()
149 idx += num_col; in fadst4x4_sse4_1()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops.cc | 92 const int64 num_col = input_flat.dimension(1); in Compute() local
129 Eigen::DSizes<Eigen::DenseIndex, 1> out_slice_shape(num_col); in Compute()
162 out_index - uninitialized_index, num_col); in Compute()
182 num_col); in Compute()
659 const int64 num_col = input_flat.dimension(1); in Compute() local
727 out_index - uninitialized_index, num_col); in Compute()
752 output_rows - uninitialized_index, num_col); in Compute()
|
| D | sparse_conditional_accumulator.h | 268 const int64 num_col = grad_flat.dimension(1); in AddToAccumGradFunction() local
270 Eigen::DSizes<Eigen::DenseIndex, 1> slice_shape(num_col); in AddToAccumGradFunction()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_kronecker.py | 44 def _unvec_by(y, num_col): argument
50 [array_ops.shape(y)[:-1], [num_col, -1]], axis=0)))
|