Searched defs:partial_shape (Results 1 – 9 of 9) sorted by relevance
156 PartialTensorShape partial_shape(shape); in RewriteIntoTempVariable() local
696 PartialTensorShape partial_shape; in OptimizeGraph() local
81 PartialTensorShape partial_shape; in TryGetElementShapeFromInput() local
859 const PartialTensorShape& partial_shape, ShapeHandle* out) { in MakeShapeFromPartialTensorShape()884 PartialTensorShape partial_shape(proto); in MakeShapeFromShapeProto() local
658 PartialTensorShape partial_shape; in FakeParamOp() local
658 const Shape& partial_shape = rhs->operand(1)->shape(); in HandleAdd() local
645 Shape partial_shape = ShapeUtil::MakeShape(F32, {1, 12, 512}); in TEST_F() local
2153 const int64_t partial_shape[] = {17, -1}; in TEST_F() local
5811 PartialTensorShape partial_shape; in ConvertSegmentToGraphDef() local