Searched defs:reshaped_shape (Results 1 – 11 of 11) 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
200 TensorShape reshaped_shape = _itransformed_output.info()->tensor_shape(); in configure() local
146 Shape reshaped_shape = in RewriteReduction() local
207 TensorShape reshaped_shape = _itransformed_output.info()->tensor_shape(); in configure() local
171 …const TensorShape reshaped_shape = TensorShape(nhwc_weights_shape[0], nhwc_weights_shape[1] * nhwc… in validate() local
1010 auto reshaped_shape = ValuesToRank1( in matchAndRewrite() local1115 std::vector<int64_t> reshaped_shape; in matchAndRewrite() local
3971 Shape reshaped_shape = reshape->shape(); in HandleReshape() local
5969 Shape reshaped_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
605 Shape reshaped_shape = in AddBroadcastSequence() local