Home
last modified time | relevance | path

Searched refs:input_matrix (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_ordering_amd_op.cc68 const CSRSparseMatrix* input_matrix; in Compute() local
69 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &input_matrix)); in Compute()
71 const Tensor& dense_shape = input_matrix->dense_shape(); in Compute()
86 const int batch_size = input_matrix->batch_size(); in Compute()
96 10 * num_rows * (input_matrix->total_nnz() / batch_size); in Compute()
107 num_rows, num_rows, input_matrix->nnz(batch_index), in Compute()
108 input_matrix->row_pointers_vec(batch_index).data(), in Compute()
109 input_matrix->col_indices_vec(batch_index).data(), in Compute()
110 input_matrix->col_indices_vec(batch_index).data()); in Compute()
Dtranspose_op.cc102 const CSRSparseMatrix* input_matrix; in Compute() local
103 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &input_matrix)); in Compute()
105 ctx, input_matrix->dtype() == DataTypeToEnum<T>::value, in Compute()
107 DataTypeString(input_matrix->dtype()), " vs. ", in Compute()
114 transpose(ctx, conjugate_, *input_matrix, &output_matrix)); in Compute()
151 OpKernelContext* ctx, bool conjugate, const CSRSparseMatrix& input_matrix, in operator ()() argument
153 const int rank = input_matrix.dims(); in operator ()()
155 const Tensor& input_dense_shape_t = input_matrix.dense_shape(); in operator ()()
158 const int64 batch_size = input_matrix.batch_size(); in operator ()()
167 Tensor batch_ptr_t = input_matrix.batch_pointers(); in operator ()()
[all …]
Dsparse_cholesky_op.cc79 const CSRSparseMatrix* input_matrix; in Compute() local
80 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &input_matrix)); in Compute()
85 ValidateInputs(ctx, *input_matrix, input_permutation_indices, &batch_size, in Compute()
104 (input_matrix->total_nnz() / batch_size) / num_rows; in Compute()
118 num_rows, num_rows, input_matrix->nnz(batch_index), in Compute()
119 input_matrix->row_pointers_vec(batch_index).data(), in Compute()
120 input_matrix->col_indices_vec(batch_index).data(), in Compute()
121 input_matrix->values_vec<T>(batch_index).data()); in Compute()
216 DataTypeToEnum<T>::value, input_matrix->dense_shape(), batch_ptr, in Compute()
Dkernels.h211 const CSRSparseMatrix& input_matrix,
/external/tensorflow/tensorflow/python/kernel_tests/
Dfractional_avg_pool_op_test.py42 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)
Dfractional_max_pool_op_test.py42 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)
Ddiag_op_test.py882 input_matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
884 result_eval = result.eval(feed_dict={matrix: input_matrix})
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dreconstruction_ops_test.py151 input_matrix = np.squeeze(self.powers[0, :, :])
152 input_matrix = input_matrix[np.newaxis, :, :].astype(float)
153 signal = constant_op.constant(input_matrix, dtype=dtypes.float32)
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_requantization_range_per_channel_op.cc66 auto input_matrix = input.flat_inner_dims<qint32>(); in Compute() local
74 auto transposed_input = input_matrix.shuffle(shuffling); in Compute()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op.cc134 ConstMatrixMap input_matrix( in ComputeTensorSlice() local
148 lu_decomposition(input_matrix); in ComputeTensorSlice()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1136 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in FullyConnected()
1145 gemmlowp_context, filter_matrix, input_matrix, &output_matrix, in FullyConnected()
1694 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in FullyConnected()
1717 gemmlowp_context, weights_matrix, input_matrix, &output_matrix, in FullyConnected()
2153 gemmlowp::MatrixMap<const int8, gemmlowp::MapOrder::ColMajor> input_matrix( in FullyConnected()
2163 gemmlowp_context, filter_matrix, input_matrix, &output_matrix, in FullyConnected()
2716 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in Conv()
2725 gemmlowp_context, filter_matrix, input_matrix, &output_matrix, in Conv()
2894 gemmlowp::MatrixMap<const uint8, gemmlowp::MapOrder::ColMajor> input_matrix( in ConvAsGemm()
2903 gemmlowp_context, filter_matrix, input_matrix, &output_matrix, in ConvAsGemm()
[all …]