Home
last modified time | relevance | path

Searched defs:HandlePad (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_normalization.cc383 Status HandlePad(HloInstruction* hlo) override { in HandlePad() function in xla::__anon52bdec210111::LayoutNormalizationVisitor
H A Ddfs_hlo_visitor_with_default.h211 Status HandlePad(HloInstructionPtr pad) override { in HandlePad() function
H A Dhlo_cost_analysis.cc485 Status HloCostAnalysis::HandlePad(const HloInstruction*) { return OkStatus(); } in HandlePad() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc1347 Status ShapeVerifier::HandlePad(HloInstruction* pad) { in HandlePad() function in xla::ShapeVerifier
H A Ddynamic_dimension_inference.cc390 Status DynamicDimensionInferenceVisitor::HandlePad(HloInstruction* hlo) { in HandlePad() function in xla::DynamicDimensionInferenceVisitor
H A Dalgebraic_simplifier.cc3701 Status AlgebraicSimplifierVisitor::HandlePad(HloInstruction* pad) { in HandlePad() function in xla::AlgebraicSimplifierVisitor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_space_to_depth_pass.cc113 LogicalResult HandlePad(TF::PadOp op, int32_t kernel_size, int32_t block_size) { in HandlePad() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc2115 Status IrEmitter::HandlePad(HloInstruction* pad) { in HandlePad() function in xla::cpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc3378 Status SpmdPartitioningVisitor::HandlePad(HloInstruction* hlo) { in HandlePad() function in xla::spmd::SpmdPartitioningVisitor