| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 208 Status HloCostAnalysis::HandleParameter(const HloInstruction*) { in HandleParameter()
216 Status HloCostAnalysis::HandleConstant(const HloInstruction*) { in HandleConstant()
224 Status HloCostAnalysis::HandleIota(const HloInstruction*) { in HandleIota()
244 Status HloCostAnalysis::HandleTupleSelect(const HloInstruction*) { in HandleTupleSelect()
248 Status HloCostAnalysis::HandleReverse(const HloInstruction*) { in HandleReverse()
299 Status HloCostAnalysis::HandleConcatenate(const HloInstruction*) { in HandleConcatenate()
307 Status HloCostAnalysis::HandleCopy(const HloInstruction*) { in HandleCopy()
471 Status HloCostAnalysis::HandleBitcast(const HloInstruction*) { in HandleBitcast()
480 Status HloCostAnalysis::HandleBroadcast(const HloInstruction*) { in HandleBroadcast()
484 Status HloCostAnalysis::HandlePad(const HloInstruction*) { in HandlePad()
[all …]
|
| D | transpose_folding.h | 38 static OperandIndices NeverFoldTranspose(const HloInstruction&, in NeverFoldTranspose()
44 static OperandIndices AlwaysFoldTranspose(const HloInstruction&, in AlwaysFoldTranspose()
|
| D | logical_buffer_analysis.cc | 113 Status LogicalBufferAnalysis::HandleGetTupleElement(HloInstruction*) { in HandleGetTupleElement() argument
132 Status LogicalBufferAnalysis::HandleBitcast(HloInstruction*) { in HandleBitcast() argument
138 Status LogicalBufferAnalysis::HandleDomain(HloInstruction*) { in HandleDomain() argument
|
| D | hlo_computation.cc | 384 absl::flat_hash_map<HloInstruction*, VisitState>* visited) const { in ComputeInstructionPostOrder()
870 auto operands_eq = [](const HloInstruction*, const HloInstruction*) { in EqualInternal()
989 std::pair<const HloInstruction*, std::unique_ptr<HloInstruction>> r1, in CloneWithReplacementPairs()
999 std::pair<const HloInstruction*, std::unique_ptr<HloInstruction>> r1, in CloneWithReplacementPairs()
1011 std::pair<const HloInstruction*, std::unique_ptr<HloInstruction>> r1, in CloneWithReplacementPairs()
1025 absl::flat_hash_map<const HloInstruction*, std::unique_ptr<HloInstruction>> in CloneWithReplacements()
|
| D | hlo_domain_metadata.h | 32 class HloInstruction; variable
|
| D | hlo_casting_utils.h | 26 class HloInstruction; variable
|
| D | map_inliner.cc | 42 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() argument
|
| D | dynamic_dimension_inference.cc | 228 int64 operand_index, HloInstruction* dynamic_size) { in HandleTuple() argument
639 HloInstruction*) { in HandleGetDimensionSize() argument
1120 int64 /*operand_index*/, HloInstruction* dynamic_size) { in HandleDynamicSlice() argument
1144 int64 operand_index, HloInstruction* dynamic_size) { in HandleDynamicUpdateSlice() argument
1175 int64 /*operand_index*/, HloInstruction* dynamic_size) { in HandleReverse() argument
1283 HloInstruction* dynamic_size) -> Status { in HandleConditional() argument
1325 int64 operand_index, HloInstruction* dynamic_size) { in HandleConditional() argument
1408 int64 operand_index, HloInstruction* operand_dynamic_size) { in HandleScatter() argument
1443 HloInstruction* dynamic_size) { in HandleWhile() argument
1471 int64 operand_index, HloInstruction* dynamic_size) -> Status { in HandleWhile() argument
|
| D | hlo_clone_context.h | 26 class HloInstruction; variable
|
| D | hlo_execution_profile.h | 31 class HloInstruction; variable
|
| D | hlo_verifier.h | 115 Status FinishVisit(HloInstruction*) override { return Status::OK(); } in FinishVisit() argument
|
| D | instruction_fusion.cc | 227 absl::flat_hash_map<std::pair<HloInstruction*, HloInstruction*>, bool>* in CanFuseOnAllPaths() argument
|
| D | hlo_reachability.cc | 78 absl::FunctionRef<void(const HloInstruction*, in BuildWithRestrictions()
|
| D | while_loop_invariant_code_motion.cc | 39 flat_hash_map<HloInstruction*, HloInstruction*>* hoisted_instructions, in CreateLoopInvariantCopy()
|
| D | ar_crs_combiner.cc | 370 auto eq_operands = [](const HloInstruction*, const HloInstruction*) { in InstructionsComputeSameValue()
|
| D | while_loop_expensive_invariant_code_motion.cc | 54 flat_hash_map<HloInstruction*, InvariantInfo>* invariant_instructions, in CreateLoopInvariantCopy()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 195 Status IrEmitter::HandleSend(HloInstruction*) { in HandleSend() argument
199 Status IrEmitter::HandleSendDone(HloInstruction*) { in HandleSendDone() argument
203 Status IrEmitter::HandleRecv(HloInstruction*) { in HandleRecv() argument
207 Status IrEmitter::HandleRecvDone(HloInstruction*) { in HandleRecvDone() argument
211 Status IrEmitter::HandleScatter(HloInstruction*) { in HandleScatter() argument
598 Status IrEmitter::HandleCustomCall(HloInstruction*) { in HandleCustomCall() argument
602 Status IrEmitter::HandleInfeed(HloInstruction*) { in HandleInfeed() argument
607 Status IrEmitter::HandleOutfeed(HloInstruction*) { in HandleOutfeed() argument
612 Status IrEmitter::HandleBatchNormInference(HloInstruction*) { in HandleBatchNormInference() argument
620 Status IrEmitter::HandleBatchNormTraining(HloInstruction*) { in HandleBatchNormTraining() argument
[all …]
|
| D | thunk_schedule.cc | 31 const absl::flat_hash_map<const HloInstruction*, Thunk*>& hlo_to_thunk) { in AddDependenciesOnTransitiveOperands()
|
| D | cudnn_batchnorm_rewriter.cc | 37 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 41 StatusOr<HloInstruction*> PartitionConvolutionWithBatchGroupCount( in PartitionConvolutionWithBatchGroupCount() argument
133 StatusOr<HloInstruction*> PartitionConvolutionWithFeatureGroupCount( in PartitionConvolutionWithFeatureGroupCount() argument
227 PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
230 const std::function<StatusOr<HloInstruction*>( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() argument
514 PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
517 const std::function<StatusOr<HloInstruction*>( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() argument
740 StatusOr<HloInstruction*> PartitionConvolutionTiledOutput( in PartitionConvolutionTiledOutput() argument
816 StatusOr<HloInstruction*> PartitionConvolutionBaseCase( in PartitionConvolutionBaseCase() argument
964 StatusOr<HloInstruction*> PartitionConvolution( in PartitionConvolution() argument
|
| D | dot_handler.cc | 466 StatusOr<HloInstruction*> PartitionBaseCase( in PartitionBaseCase() argument
1648 StatusOr<HloInstruction*> PartitionDotGroupOnBatch( in PartitionDotGroupOnBatch() argument
1979 StatusOr<HloInstruction*> PartitionDotGroupOnNonContracting( in PartitionDotGroupOnNonContracting() argument
2075 StatusOr<HloInstruction*> PartitionDotGroupOnContracting( in PartitionDotGroupOnContracting() argument
2431 StatusOr<HloInstruction*> PartitionDot( in PartitionDot() argument
2798 StatusOr<HloInstruction*> PartitionDot( in PartitionDot() argument
2835 Status SpmdPartitioningVisitor::HandleDotHelper( in HandleDotHelper()
2836 HloInstruction* hlo, const DotConvDimsMapping& dims_mapping, in HandleDotHelper() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_module_importer.h | 33 class HloInstruction; variable
|
| D | hlo_function_importer.h | 40 class HloInstruction; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 226 const std::unordered_map<const HloInstruction*, int64>& in GetCandidatesForComputation()
237 std::unordered_map<const HloInstruction*, int64>* hlo_to_profile_idx, in CollectProfileCandidates()
238 const std::unordered_map<const HloInstruction*, int64>& assigned_indices) in CollectProfileCandidates()
257 Status HandleConstant(HloInstruction*) override { return Status::OK(); } in HandleConstant() argument
259 Status HandleParameter(HloInstruction*) override { return Status::OK(); } in HandleParameter() argument
539 std::unordered_map<const HloInstruction*, int64>* in CreateHloProfilingArtifacts()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_extractor.cc | 81 Status FinishVisit(const HloInstruction* /*root*/) override { in FinishVisit()
|