Searched defs:shape_shape (Results 1 – 8 of 8) sorted by relevance
48 std::initializer_list<int> shape_shape, in ReshapeOpModel()95 std::initializer_list<int> shape_shape, in BuildWithTensorShape()115 std::initializer_list<int> shape_shape, in BuildWithConstantTensorShape()
33 std::initializer_list<int> shape_shape) { in BroadcastToOpModel()45 std::initializer_list<int> shape_shape, in BroadcastToOpModel()
52 const RuntimeShape& shape_shape, in CheckShapes()
43 const TensorShape shape_shape = ctx->InputShape("shape"); in Compile() local
33 shape_shape=None): argument
962 Output shape_shape = ops::Const(root.WithOpName("shape_shape"), {2}, {1}); in TEST() local1033 Output shape_shape = in TEST() local
2319 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor()
3026 TensorShape shape_shape({var_shape.dims()}); in CreateTpuExecuteDummyArg() local