| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | wals_test.py | 183 def input_matrix(self): member in WALSMatrixFactorizationTest
260 input_fn = self.input_fn(np_matrix=self.input_matrix,
271 input_fn = self.input_fn(np_matrix=self.input_matrix,
282 input_fn = self.input_fn(np_matrix=self.input_matrix,
325 input_fn = self.input_fn(np_matrix=self.input_matrix,
331 eval_input_fn_row = self.input_fn(np_matrix=self.input_matrix,
349 eval_input_fn_col = self.input_fn(np_matrix=self.input_matrix,
392 def input_matrix(self): member in WALSMatrixFactorizaiontTestPaddedInput
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | fractional_avg_pool_op_test.py | 42 def _AvgPoolAlongRows(self, input_matrix, row_seq, overlapping): argument
55 output_image = np.zeros(input_matrix.shape[1])
62 input_matrix[row_start:row_end, :], axis=0))) # axis 0 is along row
66 def _AvgPoolAlongCols(self, input_matrix, col_seq, overlapping): argument
79 input_matrix = input_matrix.transpose()
80 output_matrix = self._AvgPoolAlongRows(input_matrix, col_seq, overlapping)
|
| D | fractional_max_pool_op_test.py | 42 def _MaxPoolAlongRows(self, input_matrix, row_seq, overlapping): argument
55 output_image = np.zeros(input_matrix.shape[1])
62 input_matrix[row_start:row_end, :], axis=0))) # axis 0 is along row
66 def _MaxPoolAlongCols(self, input_matrix, col_seq, overlapping): argument
79 input_matrix = input_matrix.transpose()
80 output_matrix = self._MaxPoolAlongRows(input_matrix, col_seq, overlapping)
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | reconstruction_ops_test.py | 186 input_matrix = np.squeeze(self.powers[0, :, :])
187 input_matrix = input_matrix[np.newaxis, :, :].astype(float)
188 signal = constant_op.constant(input_matrix, dtype=dtypes.float32)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_requantization_range_per_channel_op.cc | 64 auto input_matrix = input.flat_inner_dims<qint32>(); in Compute() local
72 auto transposed_input = input_matrix.shuffle(shuffling); in Compute()
|
| D | lu_op.cc | 134 ConstMatrixMap input_matrix( in ComputeTensorSlice() local
148 lu_decomposition(input_matrix); in ComputeTensorSlice()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1193 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in FullyConnected()
1202 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
1293 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in FullyConnected()
1316 gemm_context, weights_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
2438 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in Conv()
2447 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in Conv()
2624 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in ConvAsGemm()
2633 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in ConvAsGemm()
3894 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in LstmCell()
3912 gemm_context, weights_matrix, input_matrix, &output_matrix, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1212 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in FullyConnected()
1221 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
1285 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in FullyConnected()
1308 gemm_context, weights_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
2456 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in Conv()
2465 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in Conv()
3778 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in LstmCell()
3796 gemm_context, weights_matrix, input_matrix, &output_matrix, in LstmCell()
|
| D | legacy_optimized_ops.h | 769 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in ConvAsGemm()
778 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in ConvAsGemm()
|