Searched defs:shape_n (Results 1 – 3 of 3) sorted by relevance
98 const int64_t shape_n = ShapeUtil::TupleElementCount(shape); in MakePartitionedShape() local120 const int64_t shape_n = ShapeUtil::TupleElementCount(shape); in MakeNonPaddedShapeForGivenPartition() local
452 auto shape_n = ops::ShapeN(s.WithOpName("shape_n"), {recv0, recv1}); in TEST_F() local
702 def shape_n(input, out_type=dtypes.int32, name=None): function