Searched defs:input_matrix (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | fractional_avg_pool_op_test.py | 39 def _AvgPoolAlongRows(self, input_matrix, row_seq, overlapping): argument
63 def _AvgPoolAlongCols(self, input_matrix, col_seq, overlapping): argument
|
| H A D | fractional_max_pool_op_test.py | 39 def _MaxPoolAlongRows(self, input_matrix, row_seq, overlapping): argument
63 def _MaxPoolAlongCols(self, input_matrix, col_seq, overlapping): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | transpose_op.cc | 102 const CSRSparseMatrix* input_matrix; in Compute() local
148 OpKernelContext* ctx, bool conjugate, const CSRSparseMatrix& input_matrix, in operator ()()
|
| H A D | sparse_ordering_amd_op.cc | 68 const CSRSparseMatrix* input_matrix; in Compute() local
|
| H A D | sparse_cholesky_op.cc | 81 const CSRSparseMatrix* input_matrix; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_requantization_range_per_channel_op.cc | 80 auto input_matrix = input.flat_inner_dims<qint32>(); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | lu_op.cc | 134 ConstMatrixMap input_matrix( in ComputeTensorSlice() local
|