Searched defs:ExpandDims (Results 1 – 5 of 5) sorted by relevance
153 Status ExpandDims(AbstractContext* ctx, AbstractTensorHandle* const input, in ExpandDims() function
140 TensorShape ExpandDims(const TensorShape& tensorShape, int axis) in ExpandDims() function
1339 class ExpandDims class
2827 TEST_F(OpTest, ExpandDims) { in TEST_F() argument
15084 func ExpandDims(scope *Scope, input tf.Output, axis tf.Output) (output tf.Output) { func