Searched refs:ScalarShape (Results 1 – 25 of 43) sorted by relevance
12
30 return shape_inference::ScalarShape(c); in __anon4afff42d0102()43 return shape_inference::ScalarShape(c); in __anon4afff42d0202()56 return shape_inference::ScalarShape(c); in __anon4afff42d0302()68 .SetShapeFn(shape_inference::ScalarShape);78 .SetShapeFn(shape_inference::ScalarShape);89 return shape_inference::ScalarShape(c); in __anon4afff42d0402()101 return shape_inference::ScalarShape(c); in __anon4afff42d0502()116 .SetShapeFn(shape_inference::ScalarShape);125 .SetShapeFn(shape_inference::ScalarShape);134 .SetShapeFn(shape_inference::ScalarShape);[all …]
42 .SetShapeFn(shape_inference::ScalarShape); // TODO(mrry): Validate that54 .SetShapeFn(shape_inference::ScalarShape); // TODO(mrry): Validate that the67 .SetShapeFn(shape_inference::ScalarShape);84 .SetShapeFn(shape_inference::ScalarShape);92 .SetShapeFn(shape_inference::ScalarShape);100 .SetShapeFn(shape_inference::ScalarShape);111 return shape_inference::ScalarShape(c); in __anon42b7b0c00102()123 return shape_inference::ScalarShape(c); in __anon42b7b0c00202()135 return shape_inference::ScalarShape(c); in __anon42b7b0c00302()148 .SetShapeFn(shape_inference::ScalarShape);[all …]
79 .SetShapeFn(shape_inference::ScalarShape);88 .SetShapeFn(shape_inference::ScalarShape);100 .SetShapeFn(shape_inference::ScalarShape);108 .SetShapeFn(shape_inference::ScalarShape);116 .SetShapeFn(shape_inference::ScalarShape)122 .SetShapeFn(shape_inference::ScalarShape);
204 .SetShapeFn(shape_inference::ScalarShape);224 .SetShapeFn(shape_inference::ScalarShape);244 .SetShapeFn(shape_inference::ScalarShape);264 .SetShapeFn(shape_inference::ScalarShape);383 .SetShapeFn(shape_inference::ScalarShape);388 .SetShapeFn(shape_inference::ScalarShape);405 .SetShapeFn(shape_inference::ScalarShape);459 .SetShapeFn(shape_inference::ScalarShape);737 return shape_inference::ScalarShape(c); in __anonea01102b1802()826 return shape_inference::ScalarShape(c); in __anonea01102b1b02()[all …]
34 .SetShapeFn(shape_inference::ScalarShape);75 .SetShapeFn(shape_inference::ScalarShape);
72 .SetShapeFn(shape_inference::ScalarShape);80 .SetShapeFn(shape_inference::ScalarShape);
45 .SetShapeFn(tensorflow::shape_inference::ScalarShape);59 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
273 .SetShapeFn(shape_inference::ScalarShape);290 .SetShapeFn(shape_inference::ScalarShape);314 .SetShapeFn(shape_inference::ScalarShape);331 .SetShapeFn(shape_inference::ScalarShape);353 .SetShapeFn(shape_inference::ScalarShape);
26 .SetShapeFn(shape_inference::ScalarShape);
33 .SetShapeFn(shape_inference::ScalarShape);
66 .SetShapeFn(shape_inference::ScalarShape);
260 .SetShapeFn(shape_inference::ScalarShape);318 .SetShapeFn(shape_inference::ScalarShape);
25 .SetShapeFn(shape_inference::ScalarShape);
96 .SetShapeFn(shape_inference::ScalarShape);
32 .SetShapeFn(tensorflow::shape_inference::ScalarShape)45 .SetShapeFn(tensorflow::shape_inference::ScalarShape)63 .SetShapeFn(tensorflow::shape_inference::ScalarShape)89 .SetShapeFn(tensorflow::shape_inference::ScalarShape)104 .SetShapeFn(tensorflow::shape_inference::ScalarShape)121 .SetShapeFn(tensorflow::shape_inference::ScalarShape)139 .SetShapeFn(tensorflow::shape_inference::ScalarShape)152 .SetShapeFn(tensorflow::shape_inference::ScalarShape)165 .SetShapeFn(tensorflow::shape_inference::ScalarShape)228 .SetShapeFn(tensorflow::shape_inference::ScalarShape)[all …]
37 return tensorflow::shape_inference::ScalarShape(c); in __anonc746d7b60102()58 return tensorflow::shape_inference::ScalarShape(c); in __anonc746d7b60202()
27 .SetShapeFn(shape_inference::ScalarShape);42 .SetShapeFn(shape_inference::ScalarShape);51 return shape_inference::ScalarShape(c); in __anonc329d32f0202()63 .SetShapeFn(shape_inference::ScalarShape);675 .SetShapeFn(shape_inference::ScalarShape);701 .SetShapeFn(shape_inference::ScalarShape);
28 .SetShapeFn(shape_inference::ScalarShape)
30 .SetShapeFn(shape_inference::ScalarShape);
48 TEST(ShapeDebugString, ScalarShape) { TestShapeMatch(TensorShape({})); } in TEST() argument
34 func ScalarShape() Shape { func
33 shape: ScalarShape(),
111 return ScalarShape()
31 Placeholder(s.SubScope("y"), tf.Float, PlaceholderShape(tf.ScalarShape()))
77 variable = VarHandleOp(s, tf.Int32, tf.ScalarShape())