Searched refs:tensorshape_proto (Results 1 – 1 of 1) sorted by relevance
2715 TensorProto tensorshape_proto; in CreatePerHostDummyArgs() local2716 tensorshape_proto.set_dtype(DT_INT32); in CreatePerHostDummyArgs()2718 tensorshape_proto.add_int_val(var_shape.dim_size(i)); in CreatePerHostDummyArgs()2721 shape_shape.AsProto(tensorshape_proto.mutable_tensor_shape()); in CreatePerHostDummyArgs()2722 AddNodeAttr("value", tensorshape_proto, &shape_tensor_def); in CreatePerHostDummyArgs()