Home
last modified time | relevance | path

Searched defs:HandleSelectAndScatter (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddfs_hlo_visitor_with_default.h202 Status HandleSelectAndScatter(HloInstructionPtr select_and_scatter) override { in HandleSelectAndScatter() function
H A Dhlo_cost_analysis.cc438 Status HloCostAnalysis::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc1280 Status ShapeVerifier::HandleSelectAndScatter(HloInstruction* instruction) { in HandleSelectAndScatter() function in xla::ShapeVerifier
H A Ddynamic_dimension_inference.cc1271 Status DynamicDimensionInferenceVisitor::HandleSelectAndScatter( in HandleSelectAndScatter() function in xla::DynamicDimensionInferenceVisitor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc687 Status IrEmitter::HandleSelectAndScatter(HloInstruction* select_and_scatter) { in HandleSelectAndScatter() function in xla::cpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc3955 Status SpmdPartitioningVisitor::HandleSelectAndScatter(HloInstruction* hlo) { in HandleSelectAndScatter() function in xla::spmd::SpmdPartitioningVisitor