Searched defs:reshaped_shape (Results 1 – 11 of 11) sorted by relevance
72 std::vector<int64_t> reshaped_shape; in Compile() local
70 std::vector<int64_t> reshaped_shape(input_rank + block_rank); in BatchToSpace() local
201 TensorShape reshaped_shape = _itransformed_output.info()->tensor_shape(); in configure() local
197 Shape reshaped_shape = in RewriteReduction() local
210 TensorShape reshaped_shape = _itransformed_output.info()->tensor_shape(); in configure() local
159 …const TensorShape reshaped_shape = TensorShape(nhwc_weights_shape[0], nhwc_weights_shape[1] * nhwc… in validate() local
1000 auto reshaped_shape = ValuesToRank1( in matchAndRewrite() local1105 SmallVector<int64_t> reshaped_shape; in matchAndRewrite() local
4256 Shape reshaped_shape = reshape->shape(); in HandleReshape() local
6581 Shape reshaped_shape = ShapeUtil::MakeShape(F32, {0}); in TEST_F() local
658 Shape reshaped_shape = in AddBroadcastSequence() local