Home
last modified time | relevance | path

Searched refs:select_and_scatter (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1732 HloInstruction* select_and_scatter) { in HandleSelectAndScatter() argument
1733 CHECK_EQ(select_and_scatter->operand_count(), 3); in HandleSelectAndScatter()
1734 const auto* operand = select_and_scatter->operand(0); in HandleSelectAndScatter()
1735 const auto* source = select_and_scatter->operand(1); in HandleSelectAndScatter()
1736 const Window& window = select_and_scatter->window(); in HandleSelectAndScatter()
1744 thunks.emplace_back(BuildKernelThunk(select_and_scatter)); in HandleSelectAndScatter()
1746 select_and_scatter, static_cast<KernelThunk*>(thunks.back().get()))); in HandleSelectAndScatter()
1748 thunks.emplace_back(BuildKernelThunk(select_and_scatter)); in HandleSelectAndScatter()
1750 MakeUnique<SequentialThunk>(std::move(thunks), select_and_scatter)); in HandleSelectAndScatter()
1797 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "inner"), in HandleSelectAndScatter()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h155 Status HandleSelectAndScatter(HloInstructionPtr select_and_scatter) override { in HandleSelectAndScatter() argument
156 return DefaultAction(select_and_scatter); in HandleSelectAndScatter()
Dhlo_evaluator.cc1524 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override { in HandleSelectAndScatter() argument
1525 auto operand = select_and_scatter->operand(0); in HandleSelectAndScatter()
1526 auto source = select_and_scatter->operand(1); in HandleSelectAndScatter()
1527 const Window& window = select_and_scatter->window(); in HandleSelectAndScatter()
1530 parent_->GetEvaluatedLiteralFor(select_and_scatter->operand(2)); in HandleSelectAndScatter()
1534 auto result = Literal::CreateFromShape(select_and_scatter->shape()); in HandleSelectAndScatter()
1549 HloComputation* select = select_and_scatter->select(); in HandleSelectAndScatter()
1550 HloComputation* scatter = select_and_scatter->scatter(); in HandleSelectAndScatter()
1625 parent_->evaluated_[select_and_scatter] = std::move(result); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc614 Status IrEmitter::HandleSelectAndScatter(HloInstruction* select_and_scatter) { in HandleSelectAndScatter() argument
615 CHECK_EQ(select_and_scatter->operand_count(), 3); in HandleSelectAndScatter()
616 const auto operand = select_and_scatter->operand(0); in HandleSelectAndScatter()
617 const auto source = select_and_scatter->operand(1); in HandleSelectAndScatter()
618 const auto init_value = select_and_scatter->operand(2); in HandleSelectAndScatter()
619 const Window& window = select_and_scatter->window(); in HandleSelectAndScatter()
633 FindOrDie(emitted_functions_, select_and_scatter->select()); in HandleSelectAndScatter()
635 FindOrDie(emitted_functions_, select_and_scatter->scatter()); in HandleSelectAndScatter()
659 select_and_scatter, /*desc=*/IrName(select_and_scatter, "init"), in HandleSelectAndScatter()
666 llvm_ir::ForLoopNest source_loops(IrName(select_and_scatter), &ir_builder_); in HandleSelectAndScatter()
[all …]
Dir_emitter.h135 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override;
/external/tensorflow/tensorflow/compiler/xla/tests/
Dselect_and_scatter_test.cc83 auto select_and_scatter = builder_.SelectAndScatter( in XLA_TEST_P() local
88 ComputeAndCompare(&builder_, select_and_scatter, {}, ErrorSpec(1e-5)); in XLA_TEST_P()