Searched defs:reshaped_shape (Results 1 – 8 of 8) sorted by relevance
72 std::vector<int64> reshaped_shape; in Compile() local
70 std::vector<int64> reshaped_shape(input_rank + block_rank); in BatchToSpace() local
150 Shape reshaped_shape = in RewriteReduction() local
970 auto reshaped_shape = ValuesToRank1( in matchAndRewrite() local1077 std::vector<int64_t> reshaped_shape; in matchAndRewrite() local
3880 Shape reshaped_shape = reshape->shape(); in HandleReshape() local
5880 Shape reshaped_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
595 Shape reshaped_shape = in AddBroadcastSequence() local