Home
last modified time | relevance | path

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

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