Home
last modified time | relevance | path

Searched refs:reduce_window (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dtree_reduction_rewriter.cc83 HloInstruction *reduce_window = in HandleReduce() local
89 HloInstruction::CreateReduce(reduce_shape, reduce_window, initial_value, in HandleReduce()
Dhlo_cost_analysis.cc410 const HloInstruction* reduce_window) { in HandleReduceWindow() argument
411 const Window& window = reduce_window->window(); in HandleReduceWindow()
412 auto function = reduce_window->to_apply(); in HandleReduceWindow()
425 ShapeUtil::ElementsIn(reduce_window->shape().IsArray() in HandleReduceWindow()
426 ? reduce_window->shape() in HandleReduceWindow()
427 : reduce_window->shape().tuple_shapes(0)); in HandleReduceWindow()
Ddfs_hlo_visitor_with_default.h186 Status HandleReduceWindow(HloInstructionPtr reduce_window) override { in HandleReduceWindow() argument
187 return DefaultAction(reduce_window); in HandleReduceWindow()
Dalgebraic_simplifier.cc345 Status HandleReduceWindow(HloInstruction* reduce_window) override;
4764 HloInstruction* reduce_window) { in HandleReduceWindow() argument
4766 if (reduce_window->shape().IsTuple()) { in HandleReduceWindow()
4769 if (ShapeUtil::IsZeroElementArray(reduce_window->operand(0)->shape())) { in HandleReduceWindow()
4771 reduce_window, in HandleReduceWindow()
4772 HloInstruction::CreateBroadcast(reduce_window->shape(), in HandleReduceWindow()
4773 reduce_window->mutable_operand(1), {})); in HandleReduceWindow()
4775 auto operand = reduce_window->mutable_operand(0); in HandleReduceWindow()
4776 const Window& window = reduce_window->window(); in HandleReduceWindow()
4777 auto function = reduce_window->to_apply(); in HandleReduceWindow()
[all …]
Dconvolution_group_converter.cc397 auto reduce_window = add(HloInstruction::CreateReduceWindow( in HandleBatchGroupCount() local
401 Shape convert_back_shape = reduce_window->shape(); in HandleBatchGroupCount()
406 HloInstruction::CreateConvert(convert_back_shape, reduce_window); in HandleBatchGroupCount()
Dhlo_verifier.cc892 Status ShapeVerifier::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow() argument
893 VLOG(2) << "Verify reduce window:" << reduce_window->ToString() << "\n"; in HandleReduceWindow()
894 auto reduce_window_instr = Cast<HloReduceWindowInstruction>(reduce_window); in HandleReduceWindow()
898 VLOG(2) << "reduce instruction is :" << reduce_window->ToString() << "\n"; in HandleReduceWindow()
900 reduce_window, ShapeInference::InferReduceWindowShape( in HandleReduceWindow()
901 input_shapes, init_shapes, reduce_window->window(), in HandleReduceWindow()
902 reduce_window->to_apply()->ComputeProgramShape()))); in HandleReduceWindow()
906 : SameElementTypesForOperandsAndToApplyParameters(*reduce_window, in HandleReduceWindow()
Dhlo_cost_analysis.h110 Status HandleReduceWindow(const HloInstruction* reduce_window) override;
Delemental_ir_emitter.cc2590 const HloReduceWindowInstruction* reduce_window, in EmitElementalReduceWindow() argument
2603 int64 input_count = reduce_window->input_count(); in EmitElementalReduceWindow()
2608 auto operand = reduce_window->input_arrays()[operand_index]; in EmitElementalReduceWindow()
2632 const Window& window = reduce_window->window(); in EmitElementalReduceWindow()
2633 llvm_ir::ForLoopNest loops(IrName(reduce_window), b_, index_type); in EmitElementalReduceWindow()
2677 reduce_window->input_arrays()[0]->shape().dimensions(i)))); in EmitElementalReduceWindow()
2685 std::vector<llvm::Value*> input_values(reduce_window->operand_count()); in EmitElementalReduceWindow()
2687 input_multi_index, reduce_window->input_arrays()[0]->shape(), index_type); in EmitElementalReduceWindow()
2695 EmitThreadLocalCall(*reduce_window->to_apply(), in EmitElementalReduceWindow()
2705 reduce_window->shape().IsTuple()); in EmitElementalReduceWindow()
Dhlo_verifier.h93 Status HandleReduceWindow(HloInstruction* reduce_window) override;
Delemental_ir_emitter.h252 const HloReduceWindowInstruction* reduce_window,
Dhlo_evaluator_typed_visitor.h1945 Status HandleReduceWindow(HloInstruction* reduce_window) override {
1946 auto* reduce_window_instr = Cast<HloReduceWindowInstruction>(reduce_window);
1947 const Window& window = reduce_window->window();
1948 HloComputation* function = reduce_window->to_apply();
1956 ShapeUtil::Compatible(reduce_window->shape(), inferred_return_shape))
1958 << ShapeUtil::HumanStringWithLayout(reduce_window->shape())
2057 parent_->evaluated_[reduce_window] = std::move(result);
Ddynamic_dimension_inference.cc1063 HloInstruction* reduce_window = hlo; in HandleReduceWindow() local
1065 reduce_window->window().dimensions(dimension); in HandleReduceWindow()
1076 parent_->SetDynamicSize(reduce_window, {}, dimension, dynamic_size); in HandleReduceWindow()
Dsharding_propagation_test.cc1376 auto* reduce_window = FindInstruction(module.get(), "reduce-window"); in TEST_P() local
1377 ASSERT_NE(reduce_window, nullptr); in TEST_P()
1378 EXPECT_THAT(reduce_window, op::Sharding("{devices=[2,1]0,1}")); in TEST_P()
1382 EXPECT_THAT(reduce_window->sharding(), in TEST_P()
1386 EXPECT_THAT(reduce_window->sharding(), ShardingMetadata({})); in TEST_P()
Ddynamic_dimension_inference_test.cc971 auto* reduce_window = in TEST_F() local
983 EXPECT_EQ(inference_->GetDynamicSize(reduce_window, {}, 0), size_param); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo-legalize-to-parallel-loops.mlir135 func @reduce_window(%arg: memref<112x112xf32>,
138 "lmhlo.reduce_window"(%arg, %init, %result) ( {
150 // CHECK-LABEL: func @reduce_window(
/external/tensorflow/tensorflow/compiler/tests/
Dreduce_window_test.py38 output = xla.reduce_window(placeholder, init, reducer, **kwargs)
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1193 TF_ASSIGN_OR_RETURN(auto reduce_window, in EmitReduceWindowOp()
1206 reduce_window.window_dimensionsAttr(builder_.getI64TensorAttr(dims)); in EmitReduceWindowOp()
1208 reduce_window.window_stridesAttr(builder_.getI64TensorAttr(strides)); in EmitReduceWindowOp()
1211 reduce_window.base_dilationsAttr(builder_.getI64TensorAttr(base_dilations)); in EmitReduceWindowOp()
1214 reduce_window.window_dilationsAttr( in EmitReduceWindowOp()
1219 reduce_window.paddingAttr(DenseIntElementsAttr::get( in EmitReduceWindowOp()
1225 *hlo_reduce_window->called_computations()[0], &reduce_window.body(), in EmitReduceWindowOp()
1227 return reduce_window; in EmitReduceWindowOp()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py348 def reduce_window(operand, function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Delemental_ir_emitter.hlo104 // CHECK: define void @reduce_window(i8* noalias align 16 dereferenceable(29512) %[[VAL_0:.*]], i8*…
133 // CHECK: reduce_window.in_bounds-after: ; preds = %[[VAL_29:.*]], %[[VAL_…
135 // CHECK: reduce_window.in_bounds-true: ; preds = %[[VAL_30]]
263 …ROOT %reduce_window = reduce-window(%arg0, %c), window={size=1x2x2x1 stride=1x4x4x1 pad=0_0x2_0x0_…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir1320 // CHECK: "mhlo.reduce_window"(%[[ARG]], %[[INIT]])
1342 // CHECK: "mhlo.reduce_window"(%[[ARG]], %[[INIT]])
4405 // CHECK: [[DIVIDEND:%.+]] = "mhlo.reduce_window"([[CONV32]], [[ZERO]]) ( {
4429 // CHECK: [[DIVIDEND:%.+]] = "mhlo.reduce_window"([[CONV32]], [[ZERO]]) ( {
4453 // CHECK: [[DIVIDEND:%.+]] = "mhlo.reduce_window"([[CONV32]], [[ZERO]]) ( {
4477 // CHECK: [[DIVIDEND:%.+]] = "mhlo.reduce_window"([[CONV32]], [[ZERO]]) ( {
4500 // CHECK: %[[DIVIDEND:.*]] = "mhlo.reduce_window"(%[[ARG0]], %[[ZERO]]) ( {
4510 // CHECK: %[[DIVISOR:.*]] = "mhlo.reduce_window"(%[[ONES]], %[[ZERO]]) ( {
4530 // CHECK: %[[DIVIDEND:.*]] = "mhlo.reduce_window"(%[[ARG0]], %[[ZERO]]) ( {
4540 // CHECK: %[[DIVISOR:.*]] = "mhlo.reduce_window"(%[[ONES]], %[[ZERO]]) ( {
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h149 Status HandleReduceWindow(HloInstruction* reduce_window) override;
Dir_emitter.cc647 Status IrEmitter::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow() argument
661 return DefaultAction(reduce_window); in HandleReduceWindow()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlegalize_hlo.mlir1748 %2 = "mhlo.reduce_window"(%arg0, %0) ( {
1770 %2 = "mhlo.reduce_window"(%arg0, %1) ( {
1780 %3 = "mhlo.reduce_window"(%0, %1) ( {
1802 %2 = "mhlo.reduce_window"(%arg0, %0) ( {
1824 %2 = "mhlo.reduce_window"(%arg0, %1) ( {
1834 %3 = "mhlo.reduce_window"(%0, %1) ( {
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_ops.td219 def LHLO_ReduceWindowOp: LHLO_Op<"reduce_window", []>, BASE_HLO_ReduceWindowOp {
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt537 // CHECK: "lmhlo.reduce_window"(%arg0, %{{.*}}, %{{.*}}) ( {

12