Lines Matching refs:TensorInfo

30 		Inputs: map[string]*corepb.TensorInfo{
31 			"input_1": &corepb.TensorInfo{
44 			"input_2": &corepb.TensorInfo{
58 		Outputs: map[string]*corepb.TensorInfo{
59 			"output_1": &corepb.TensorInfo{
72 			"output_2": &corepb.TensorInfo{
90 		Inputs: map[string]TensorInfo{
91 			"input_1": TensorInfo{
96 			"input_2": TensorInfo{
102 		Outputs: map[string]TensorInfo{
103 			"output_1": TensorInfo{
108 			"output_2": TensorInfo{
143 	got := tensorInfoFromProto(&corepb.TensorInfo{
156 	want := TensorInfo{
171 func diffTensorInfos(a, b TensorInfo) (string, error) {  argument