Home
last modified time | relevance | path

Searched refs:HandlePad (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h114 Status HandlePad(const HloInstruction* pad) override;
Ddfs_hlo_visitor_with_default.h198 Status HandlePad(HloInstructionPtr pad) override { in HandlePad() function
Dhlo_verifier.h97 Status HandlePad(HloInstruction* pad) override;
Ddfs_hlo_visitor.h266 virtual Status HandlePad(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc107 Status HandlePad(HloInstruction* hlo) override;
359 Status DynamicDimensionInferenceVisitor::HandlePad(HloInstruction* hlo) { in HandlePad() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.cc484 Status HloCostAnalysis::HandlePad(const HloInstruction*) { in HandlePad() function in xla::HloCostAnalysis
Dhlo_verifier.cc977 Status ShapeVerifier::HandlePad(HloInstruction* pad) { in HandlePad() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc335 Status HandlePad(HloInstruction* pad) override;
3322 Status AlgebraicSimplifierVisitor::HandlePad(HloInstruction* pad) { in HandlePad() function in xla::AlgebraicSimplifierVisitor
Dhlo_evaluator_typed_visitor.h1533 Status HandlePad(HloInstruction* pad) override {
Dhlo_instruction.cc3201 return visitor->HandlePad(this); in Visit()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc158 LogicalResult HandlePad(TF::PadOp op, int32_t kernel_size, int32_t block_size) { in HandlePad() function
602 if (failed(HandlePad(pad_op, filter_shape[0], block_size))) return; in HandleFirstConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h159 Status HandlePad(HloInstruction* pad) override;
Dir_emitter.cc2004 Status IrEmitter::HandlePad(HloInstruction* pad) { in HandlePad() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h421 Status HandlePad(HloInstruction* hlo) override;
Dspmd_partitioner.cc2622 Status SpmdPartitioningVisitor::HandlePad(HloInstruction* hlo) { in HandlePad() function in xla::spmd::SpmdPartitioningVisitor