HomeSort by relevance Sort by last modified time
    Searched refs:matmul_type (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc 629 llvm::FunctionType* matmul_type = llvm::FunctionType::get( local
639 module->getOrInsertFunction(fn_name, matmul_type);
    [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
arithmetic_optimizer_test.cc 791 for (const string matmul_type : {"MatMul", "SparseMatMul", "BatchMatMul"}) {
801 if (matmul_type == "MatMul") {
803 } else if (matmul_type == "SparseMatMul") {
805 } else if (matmul_type == "BatchMatMul") {
832 if (matmul_type == "BatchMatMul") {
    [all...]

Completed in 142 milliseconds