Home
last modified time | relevance | path

Searched refs:dynamic_update_slice (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc57 bool CanUpdateDynamicSliceInPlace(HloInstruction* dynamic_update_slice, in CanUpdateDynamicSliceInPlace() argument
59 CHECK_EQ(HloOpcode::kDynamicUpdateSlice, dynamic_update_slice->opcode()); in CanUpdateDynamicSliceInPlace()
60 const HloInstruction* operand = dynamic_update_slice->operand(0); in CanUpdateDynamicSliceInPlace()
61 return assignment.HasTopLevelAllocation(dynamic_update_slice) && in CanUpdateDynamicSliceInPlace()
63 assignment.SharesTopLevelSlice(dynamic_update_slice, operand); in CanUpdateDynamicSliceInPlace()
200 auto* dynamic_update_slice = fusion->fused_expression_root(); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
202 const auto* update = dynamic_update_slice->operand(1); in EmitFusedDynamicUpdateSliceInPlaceImpl()
203 const auto* start_indices = dynamic_update_slice->operand(2); in EmitFusedDynamicUpdateSliceInPlaceImpl()
225 TF_RETURN_IF_ERROR(dynamic_update_slice->Accept(&fused_emitter)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
230 fused_emitter.GetGenerator(dynamic_update_slice->operand(2 + index)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
Ddynamic_update_slice_util.h54 bool CanUpdateDynamicSliceInPlace(HloInstruction* dynamic_update_slice,
/external/tensorflow/tensorflow/compiler/tests/
Ddynamic_slice_ops_test.py47 xla.dynamic_update_slice, [
55 xla.dynamic_update_slice, [
63 xla.dynamic_update_slice, [
73 xla.dynamic_update_slice, [
83 xla.dynamic_update_slice, [
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc267 const HloInstruction* dynamic_update_slice) { in HandleDynamicUpdateSlice() argument
269 GetShapeSize(dynamic_update_slice->operand(1)->shape()) * 2 + in HandleDynamicUpdateSlice()
270 GetShapeSize(dynamic_update_slice->operand(2)->shape()); in HandleDynamicUpdateSlice()
273 GetShapeSize(dynamic_update_slice->operand(1)->shape())); in HandleDynamicUpdateSlice()
276 1, GetShapeSize(dynamic_update_slice->operand(1)->shape())); in HandleDynamicUpdateSlice()
278 2, GetShapeSize(dynamic_update_slice->operand(2)->shape())); in HandleDynamicUpdateSlice()
Ddfs_hlo_visitor_with_default.h162 HloInstructionPtr dynamic_update_slice) override { in HandleDynamicUpdateSlice() argument
163 return DefaultAction(dynamic_update_slice); in HandleDynamicUpdateSlice()
Dhlo_evaluator_typed_visitor.h1547 HloInstruction* dynamic_update_slice) override {
1548 auto operand = dynamic_update_slice->operand(0);
1549 auto update = dynamic_update_slice->operand(1);
1550 auto start_indices = dynamic_update_slice->operand(2);
1551 auto result_shape = dynamic_update_slice->shape();
1556 Cast<HloDynamicUpdateSliceInstruction>(dynamic_update_slice)
1572 parent_->evaluated_[dynamic_update_slice],
1575 absl::MakeConstSpan(dynamic_update_slice->operands())
1580 parent_->evaluated_[dynamic_update_slice],
1583 absl::MakeConstSpan(dynamic_update_slice->operands())
[all …]
Dhlo_dataflow_analysis_test.cc2078 auto dynamic_update_slice = in TEST_F() local
2083 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in TEST_F()
2087 {dynamic_update_slice, starts, update, gte1}, in TEST_F()
2121 auto dynamic_update_slice = in TEST_F() local
2126 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in TEST_F()
2130 {dynamic_update_slice, starts, update, gte1}, in TEST_F()
2389 auto dynamic_update_slice = in TEST_F() local
2394 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in TEST_F()
2398 {dynamic_update_slice, starts, update, gte1}, in TEST_F()
2430 auto dynamic_update_slice = in TEST_F() local
[all …]
Dhlo_verifier.cc724 HloInstruction* dynamic_update_slice) { in HandleDynamicUpdateSlice() argument
726 dynamic_update_slice, in HandleDynamicUpdateSlice()
728 dynamic_update_slice->operand(0)->shape(), in HandleDynamicUpdateSlice()
729 dynamic_update_slice->operand(1)->shape(), in HandleDynamicUpdateSlice()
730 Cast<HloDynamicUpdateSliceInstruction>(dynamic_update_slice) in HandleDynamicUpdateSlice()
Dhlo_cost_analysis.h103 const HloInstruction* dynamic_update_slice) override;
Dtuple_points_to_analysis_test.cc915 auto dynamic_update_slice = in TEST_F() local
919 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in TEST_F()
923 {dynamic_update_slice, starts, update, gte1}, in TEST_F()
Dhlo_verifier.h85 HloInstruction* dynamic_update_slice) override;
Dalgebraic_simplifier.cc356 HloInstruction* dynamic_update_slice) override;
3531 HloInstruction* dynamic_update_slice) { in HandleDynamicUpdateSlice() argument
3532 auto update = dynamic_update_slice->mutable_operand(1); in HandleDynamicUpdateSlice()
3536 if (SameShape(dynamic_update_slice, update)) { in HandleDynamicUpdateSlice()
3537 return ReplaceInstruction(dynamic_update_slice, update); in HandleDynamicUpdateSlice()
3544 return ReplaceInstruction(dynamic_update_slice, in HandleDynamicUpdateSlice()
3545 dynamic_update_slice->mutable_operand(0)); in HandleDynamicUpdateSlice()
Dalgebraic_simplifier_test.cc661 HloInstruction* dynamic_update_slice = in TEST_F() local
666 full_shape, HloOpcode::kAdd, full_param, dynamic_update_slice)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2307 HloInstruction* dynamic_update_slice) { in HandleDynamicUpdateSlice() argument
2308 auto update = dynamic_update_slice->operand(1); in HandleDynamicUpdateSlice()
2309 if (ShapeUtil::IsScalar(dynamic_update_slice->shape())) { in HandleDynamicUpdateSlice()
2310 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(dynamic_update_slice)); in HandleDynamicUpdateSlice()
2311 return EmitMemcpy(*update, *dynamic_update_slice); in HandleDynamicUpdateSlice()
2312 } else if (llvm_ir::CanUpdateDynamicSliceInPlace(dynamic_update_slice, in HandleDynamicUpdateSlice()
2314 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(dynamic_update_slice)); in HandleDynamicUpdateSlice()
2315 auto operands = GetIrArraysForOperandsOf(dynamic_update_slice); in HandleDynamicUpdateSlice()
2317 operands, GetIrArrayFor(dynamic_update_slice), in HandleDynamicUpdateSlice()
2318 IrName(dynamic_update_slice, "in_place"), &b_); in HandleDynamicUpdateSlice()
[all …]
Dir_emitter.h170 HloInstruction* dynamic_update_slice) override;
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py314 dynamic_update_slice = gen_xla_ops.xla_dynamic_update_slice variable