Searched defs:reshaped (Results 1 – 7 of 7) sorted by relevance
146 xla::XlaOp reshaped = xla::Reshape(input, reshaped_shape); in Compile() local
151 xla::XlaOp reshaped = xla::Reshape(input, reshaped_shape); in Compile() local
75 xla::XlaOp reshaped = xla::Reshape(input, reshaped_shape); in BatchToSpace() local
103 Output reshaped = ops::Reshape(scope, input, new_shape); in ExpandDimsForBroadcast() local
164 auto reshaped = ops::Reshape(root.WithOpName("reshaped"), square, new_shape); in TEST_F() local
72 auto reshaped = ExpandDims(scope, vec, -1); in BroadcastMul() local
959 auto reshaped = Reshape( in ProdGrad() local