| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 230 Status HandleReduceWindow(HloInstruction* reduce_window) override;
3084 HloInstruction* reduce_window) { in HandleReduceWindow() argument
3085 if (ShapeUtil::IsZeroElementArray(reduce_window->operand(0)->shape())) { in HandleReduceWindow()
3087 reduce_window, in HandleReduceWindow()
3088 HloInstruction::CreateBroadcast(reduce_window->shape(), in HandleReduceWindow()
3089 reduce_window->mutable_operand(1), {})); in HandleReduceWindow()
3091 auto operand = reduce_window->mutable_operand(0); in HandleReduceWindow()
3092 const Window& window = reduce_window->window(); in HandleReduceWindow()
3093 auto function = reduce_window->to_apply(); in HandleReduceWindow()
3095 TF_RET_CHECK(ShapeUtil::IsScalar(reduce_window->shape())); in HandleReduceWindow()
[all …]
|
| D | dynamic_dimension_inference.cc | 356 HloInstruction* reduce_window = hlo; in HandleReduceWindow() local
358 reduce_window->window().dimensions(dimension); in HandleReduceWindow()
363 reduce_window->ToString()); in HandleReduceWindow()
366 parent_->SetDynamicSize(reduce_window, {}, dimension, dynamic_size); in HandleReduceWindow()
|
| D | dfs_hlo_visitor_with_default.h | 167 Status HandleReduceWindow(HloInstructionPtr reduce_window) override { in HandleReduceWindow() argument
168 return DefaultAction(reduce_window); in HandleReduceWindow()
|
| D | hlo_cost_analysis.cc | 303 const HloInstruction* reduce_window) { in HandleReduceWindow() argument
304 const Window& window = reduce_window->window(); in HandleReduceWindow()
305 auto function = reduce_window->to_apply(); in HandleReduceWindow()
317 ShapeUtil::ElementsIn(reduce_window->shape()); in HandleReduceWindow()
|
| D | convolution_group_converter.cc | 300 auto reduce_window = add(HloInstruction::CreateReduceWindow( in HandleBatchGroupCount() local
304 Shape convert_back_shape = reduce_window->shape(); in HandleBatchGroupCount()
309 HloInstruction::CreateConvert(convert_back_shape, reduce_window); in HandleBatchGroupCount()
|
| D | hlo_verifier.cc | 622 Status ShapeVerifier::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow() argument
624 reduce_window, in HandleReduceWindow()
626 reduce_window->operand(0)->shape(), in HandleReduceWindow()
627 reduce_window->operand(1)->shape(), reduce_window->window(), in HandleReduceWindow()
628 reduce_window->to_apply()->ComputeProgramShape()))); in HandleReduceWindow()
632 : SameElementTypesForOperandsAndToApplyParameters(*reduce_window, in HandleReduceWindow()
|
| D | hlo_evaluator_typed_visitor.h | 1961 Status HandleReduceWindow(HloInstruction* reduce_window) override { in HandleReduceWindow() argument
1962 auto operand = reduce_window->operand(0); in HandleReduceWindow()
1963 const Window& window = reduce_window->window(); in HandleReduceWindow()
1964 HloComputation* function = reduce_window->to_apply(); in HandleReduceWindow()
1968 /*operand_shape=*/reduce_window->operand(0)->shape(), in HandleReduceWindow()
1969 /*init_value=*/reduce_window->operand(1)->shape(), window, in HandleReduceWindow()
1972 ShapeUtil::Compatible(reduce_window->shape(), inferred_return_shape)) in HandleReduceWindow()
1974 << ShapeUtil::HumanStringWithLayout(reduce_window->shape()) in HandleReduceWindow()
1979 parent_->GetEvaluatedLiteralFor(reduce_window->operand(0)); in HandleReduceWindow()
1982 parent_->GetEvaluatedLiteralFor(reduce_window->operand(1)); in HandleReduceWindow()
[all …]
|
| D | hlo_cost_analysis.h | 101 Status HandleReduceWindow(const HloInstruction* reduce_window) override;
|
| D | hlo_verifier.h | 84 Status HandleReduceWindow(HloInstruction* reduce_window) override;
|
| D | dynamic_dimension_inference_test.cc | 585 auto* reduce_window = in TEST_F() local
597 EXPECT_EQ(inference_->GetDynamicSize(reduce_window, {}, 0), size_param); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 3619 HloInstruction* reduce_window = in TEST_F() local
3627 EXPECT_EQ(root, reduce_window); in TEST_F()
3705 HloInstruction* reduce_window = in TEST_F() local
3713 EXPECT_EQ(root, reduce_window); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reduce_window_test.py | 38 output = xla.reduce_window(placeholder, init, reducer, **kwargs)
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 123 const HloReduceWindowInstruction* reduce_window,
163 Status HandleReduceWindow(HloInstruction* reduce_window) override;
|
| D | ir_emitter.cc | 641 const HloReduceWindowInstruction* reduce_window, in EmitElementalReduceWindow() argument
644 const HloInstruction* operand = reduce_window->operand(0); in EmitElementalReduceWindow()
645 const Window& window = reduce_window->window(); in EmitElementalReduceWindow()
654 Store(Load(GetEmittedValueFor(reduce_window->operand(1))), in EmitElementalReduceWindow()
657 llvm_ir::ForLoopNest loops(IrName(reduce_window, "inner"), &b_); in EmitElementalReduceWindow()
720 *reduce_window->to_apply(), {Load(accumulator_address), input_value}, in EmitElementalReduceWindow()
728 Status IrEmitter::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow() argument
742 return DefaultAction(reduce_window); in HandleReduceWindow()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 350 def reduce_window(operand, function
|