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
152 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.cc195 auto output_matrix = output_matrix_t->matrix<float>(); in Compute() local
Dscatter_nd_op.cc935 auto output_matrix = in DoScatterNd() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc84 MatrixDescriptor output_matrix, AlphaType alpha, double beta, in DoGemmWithAlgorithm()
257 const MatrixDescriptor output_matrix{output_buffer, /*needs_transpose=*/false, in RunGemm() local
/external/tensorflow/tensorflow/core/kernels/linalg/
Dcholesky_grad.cc64 MatrixMap output_matrix = outputs->at(0); in ComputeMatrix() local