Home
last modified time | relevance | path

Searched refs:dynamic_slice (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dtriangular_solve.cc491 auto dynamic_slice = [&](xla::ComputationDataHandle x, in TriangularSolveLeftLooking() local
519 a_row = dynamic_slice(body_a, {zero, i}, {m, 1}); in TriangularSolveLeftLooking()
521 a_row = dynamic_slice(body_a, {i, zero}, {1, m}); in TriangularSolveLeftLooking()
529 bodyb->Sub(dynamic_slice(body_b, {i, zero}, {1, n}), b_update); in TriangularSolveLeftLooking()
532 auto a_elt = dynamic_slice(body_a, {i, i}, {1, 1}); in TriangularSolveLeftLooking()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h136 Status HandleDynamicSlice(HloInstructionPtr dynamic_slice) override { in HandleDynamicSlice() argument
137 return DefaultAction(dynamic_slice); in HandleDynamicSlice()
Dhlo_verifier.cc226 Status ShapeVerifier::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
227 return CheckShape(dynamic_slice, ShapeInference::InferDynamicSliceShape( in HandleDynamicSlice()
228 dynamic_slice->operand(0)->shape(), in HandleDynamicSlice()
229 dynamic_slice->operand(1)->shape(), in HandleDynamicSlice()
230 dynamic_slice->dynamic_slice_sizes())); in HandleDynamicSlice()
Dalgebraic_simplifier.cc165 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
1574 HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
1575 auto operand = dynamic_slice->mutable_operand(0); in HandleDynamicSlice()
1576 auto start_indices = dynamic_slice->operand(1); in HandleDynamicSlice()
1577 if (ShapeUtil::IsScalar(dynamic_slice->shape())) { in HandleDynamicSlice()
1578 return ReplaceInstruction(dynamic_slice, operand); in HandleDynamicSlice()
1582 if (IsAll(start_indices, 0) && SameShape(operand, dynamic_slice)) { in HandleDynamicSlice()
1583 return ReplaceInstruction(dynamic_slice, operand); in HandleDynamicSlice()
Dhlo_evaluator.cc1229 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override { in HandleDynamicSlice() argument
1230 auto operand = dynamic_slice->operand(0); in HandleDynamicSlice()
1231 auto start_indices = dynamic_slice->operand(1); in HandleDynamicSlice()
1232 auto result_shape = dynamic_slice->shape(); in HandleDynamicSlice()
1236 dynamic_slice->dynamic_slice_sizes())); in HandleDynamicSlice()
1251 parent_->evaluated_[dynamic_slice], in HandleDynamicSlice()
1257 parent_->evaluated_[dynamic_slice], in HandleDynamicSlice()
1263 parent_->evaluated_[dynamic_slice], in HandleDynamicSlice()
1269 parent_->evaluated_[dynamic_slice], in HandleDynamicSlice()
Dhlo_verifier.h65 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
Dhlo_cost_analysis.h89 Status HandleDynamicSlice(const HloInstruction* dynamic_slice) override;
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h139 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
Dir_emitter.cc1960 Status IrEmitter::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
1961 if (ShapeUtil::IsScalar(dynamic_slice->shape())) { in HandleDynamicSlice()
1962 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(dynamic_slice)); in HandleDynamicSlice()
1963 return EmitMemcpy(*dynamic_slice->operand(0), *dynamic_slice); in HandleDynamicSlice()
1965 return DefaultAction(dynamic_slice); in HandleDynamicSlice()