Searched refs:init_shapes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 897 auto init_shapes = reduce_window_instr->init_value_shapes(); in HandleReduceWindow() local
901 input_shapes, init_shapes, reduce_window->window(), in HandleReduceWindow()
|
| D | hlo_evaluator_test.cc | 2902 std::vector<const Shape*> init_shapes = {&init1->shape(), &init2->shape()}; in TEST_P() local
2905 input_shapes, init_shapes, window, in TEST_P()
2953 std::vector<const Shape*> init_shapes = {&init1->shape(), &init2->shape()}; in TEST_P() local
2956 input_shapes, init_shapes, window, in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2557 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowWithGeneralPadding() local
2565 init_shapes.push_back(init_shape); in ReduceWindowWithGeneralPadding()
2576 absl::MakeSpan(init_shapes), window, in ReduceWindowWithGeneralPadding()
2602 std::vector<const Shape*> operand_shapes, init_shapes; in ReduceWindowInternal() local
2609 init_shapes.push_back(init_shape); in ReduceWindowInternal()
2621 absl::MakeSpan(init_shapes), window, to_apply_shape)); in ReduceWindowInternal()
|