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()
|