Searched defs:init_shapes (Results 1 – 4 of 4) sorted by relevance
160 std::vector<TensorShape> init_shapes; in Compile() local
1234 auto init_shapes = reduce_window_instr->init_value_shapes(); in HandleReduceWindow() local
2902 std::vector<const Shape*> init_shapes = {&init1->shape(), &init2->shape()}; in TEST_P() local2953 std::vector<const Shape*> init_shapes = {&init1->shape(), &init2->shape()}; in TEST_P() local
2585 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowWithGeneralPadding() local2630 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowInternal() local