Small utility to handle runtime shapes.

tensorflower-gardener · tensorflower-gardener · commit 73a8f9666058 · 2018-06-12T09:45:48.000-07:00
PiperOrigin-RevId: 200229761
diff --git a/tensorflow/contrib/lite/kernels/internal/types.h b/tensorflow/contrib/lite/kernels/internal/types.h
@@ -121,6 +121,10 @@ class RuntimeShape {
     }
   }
 
+  inline void BuildFrom(const std::initializer_list<int> init_list) {
+    BuildFrom<const std::initializer_list<int>>(init_list);
+  }
+
   // Returns the total count of elements, that is the size when flattened into a
   // vector.
   inline int FlatSize() const {

Original file line number	Diff line number	Diff line change
`@@ -121,6 +121,10 @@ class RuntimeShape {`
`121`	`121`	`}`
`122`	`122`	`}`
`123`	`123`
	`124`	`+ inline void BuildFrom(const std::initializer_list<int> init_list) {`
	`125`	`+ BuildFrom<const std::initializer_list<int>>(init_list);`
	`126`	`+ }`
	`127`	`+`
`124`	`128`	`// Returns the total count of elements, that is the size when flattened into a`
`125`	`129`	`// vector.`
`126`	`130`	`inline int FlatSize() const {`