Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc377 HloInstruction* select_and_scatter = hlo; in HandleSelectAndScatter() local
379 select_and_scatter->window().dimensions(dimension); in HandleSelectAndScatter()
384 select_and_scatter->ToString()); in HandleSelectAndScatter()
387 parent_->SetDynamicSize(select_and_scatter, {}, dimension, in HandleSelectAndScatter()
Ddfs_hlo_visitor_with_default.h170 Status HandleSelectAndScatter(HloInstructionPtr select_and_scatter) override { in HandleSelectAndScatter() argument
171 return DefaultAction(select_and_scatter); in HandleSelectAndScatter()
Dhlo_evaluator_typed_visitor.h1856 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override { in HandleSelectAndScatter() argument
1857 auto operand = select_and_scatter->operand(0); in HandleSelectAndScatter()
1858 auto source = select_and_scatter->operand(1); in HandleSelectAndScatter()
1859 const Window& window = select_and_scatter->window(); in HandleSelectAndScatter()
1862 parent_->GetEvaluatedLiteralFor(select_and_scatter->operand(2)); in HandleSelectAndScatter()
1866 Literal result(select_and_scatter->shape()); in HandleSelectAndScatter()
1879 HloComputation* select = select_and_scatter->select(); in HandleSelectAndScatter()
1880 HloComputation* scatter = select_and_scatter->scatter(); in HandleSelectAndScatter()
1957 parent_->evaluated_[select_and_scatter] = std::move(result); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc782 HloInstruction* select_and_scatter) { in HandleSelectAndScatter() argument
783 CHECK_EQ(select_and_scatter->operand_count(), 3); in HandleSelectAndScatter()
784 const auto* operand = select_and_scatter->operand(0); in HandleSelectAndScatter()
785 const auto* source = select_and_scatter->operand(1); in HandleSelectAndScatter()
786 const Window& window = select_and_scatter->window(); in HandleSelectAndScatter()
793 BuildInitializerThunk(select_and_scatter)); in HandleSelectAndScatter()
796 thunks.push_back(BuildKernelThunk(select_and_scatter, in HandleSelectAndScatter()
799 absl::make_unique<SequentialThunk>(std::move(thunks), select_and_scatter); in HandleSelectAndScatter()
810 select_and_scatter, launch_dimensions.launch_bound(), &b_); in HandleSelectAndScatter()
851 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "inner"), &b_, in HandleSelectAndScatter()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc745 Status IrEmitter::HandleSelectAndScatter(HloInstruction* select_and_scatter) { in HandleSelectAndScatter() argument
746 CHECK_EQ(select_and_scatter->operand_count(), 3); in HandleSelectAndScatter()
747 const auto operand = select_and_scatter->operand(0); in HandleSelectAndScatter()
748 const auto source = select_and_scatter->operand(1); in HandleSelectAndScatter()
749 const auto init_value = select_and_scatter->operand(2); in HandleSelectAndScatter()
750 const Window& window = select_and_scatter->window(); in HandleSelectAndScatter()
784 select_and_scatter, /*desc=*/IrName(select_and_scatter, "init"), in HandleSelectAndScatter()
791 llvm_ir::ForLoopNest source_loops(IrName(select_and_scatter), &b_); in HandleSelectAndScatter()
810 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "window"), &b_); in HandleSelectAndScatter()
872 *select_and_scatter->select(), in HandleSelectAndScatter()
[all …]
Dir_emitter.h164 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override;
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py406 select_and_scatter = gen_xla_ops.xla_select_and_scatter variable
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py272 return xla.select_and_scatter(