Home
last modified time | relevance | path

Searched defs:output_matrix (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dtranspose_op.cc112 CSRSparseMatrix output_matrix; in Compute() local
149 CSRSparseMatrix* output_matrix) { in operator ()()
H A Dsoftmax_op.cc64 CSRSparseMatrix output_matrix; in Compute() local
H A Dsparse_mat_mul_op.cc231 const SparseMatrix& output_matrix = output_matrices[batch_idx]; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddecode_wav_op.cc77 auto output_matrix = output->matrix<float>(); in Compute() local
H A Dload_and_remap_matrix_op.cc205 auto output_matrix = output_matrix_t->matrix<float>(); in Compute() local
H A Dscatter_nd_op.cc984 auto output_matrix = in DoScatterNd() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dcholesky_grad.cc64 MatrixMap output_matrix = outputs->at(0); in ComputeMatrix() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
H A Dgemm_pattern.cc134 MatrixDescriptor rhs_matrix, MatrixDescriptor output_matrix, in CreateTfrtOps()