Searched defs:output_matrix (Results 1 – 6 of 6) sorted by relevance
59 auto output_matrix = output_tensor.matrix<float>(); in TEST_F() local95 auto output_matrix = output_tensor.matrix<float>(); in TEST_F() local192 auto output_matrix = output_tensor.matrix<float>(); in TEST_F() local279 auto output_matrix = output_tensor.matrix<float>(); in TEST_F() local
55 MatrixDescriptor output_matrix, double alpha, double beta, in DoGemm()116 MatrixDescriptor output_matrix, double alpha, in DoGemmWithAlgorithm()159 MatrixDescriptor output_matrix, double alpha, double beta, in DoGemmAutotune()395 MatrixDescriptor output_matrix, se::Stream* stream) { in ExecuteOnStream()
77 auto output_matrix = output->matrix<float>(); in Compute() local
64 MatrixMap output_matrix = outputs->at(0); in ComputeMatrix() local
196 auto output_matrix = output_matrix_t->matrix<float>(); in Compute() local
690 auto output_matrix = in DoScatterNd() local