quantize_nodes_test.cc | 171 Output a_op = const_op("a_op", {m, k}, {2}); local 175 Output mat_mul_op = MatMul(root.WithOpName("mat_mul_op"), a_op, b_op); 215 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor)); local 221 Output mat_mul_op = MatMul(root.WithOpName("mat_mul_op"), a_op, b_op); 378 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor)); local 386 Concat(root.WithOpName("concat_op"), {a_op, b_op}, shape_op); 1237 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor)); local 1282 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor)); local 1332 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor)); local [all...] |