Home
last modified time | relevance | path

Searched refs:RunOnComputation (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dsimplify_fp_conversions.cc28 StatusOr<bool> RunOnComputation(HloComputation& computation) { in RunOnComputation() function
63 TF_ASSIGN_OR_RETURN(bool comp_changed, RunOnComputation(*computation)); in Run()
Dhlo_dce.cc64 /*static*/ StatusOr<bool> HloDCE::RunOnComputation( in RunOnComputation() function in xla::HloDCE
181 RunOnComputation(computation, remove_cross_partition_collective_ops_)); in Run()
Dhlo_dce.h47 static StatusOr<bool> RunOnComputation(
Dhlo_module_dce.cc98 HloDCE::RunOnComputation( in RunWhileDCE()
Dlayout_assignment.h506 Status RunOnComputation(LayoutConstraints* constraints,
Dlayout_assignment.cc2227 Status LayoutAssignment::RunOnComputation( in RunOnComputation() function in xla::LayoutAssignment
2511 RunOnComputation(constraints, channel_layout_constraints_)); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_broadcast_folding_rewriter.cc102 static StatusOr<bool> RunOnComputation(HloComputation *computation) { in RunOnComputation() function
114 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
Dhorizontal_input_fusion.cc160 StatusOr<bool> GpuHorizontalInputFusion::RunOnComputation( in RunOnComputation() function in xla::gpu::GpuHorizontalInputFusion
173 TF_ASSIGN_OR_RETURN(changed, RunOnComputation(comp)); in Run()
Dcusolver_rewriter.cc157 StatusOr<bool> GpusolverRewriter::RunOnComputation( in RunOnComputation() function in xla::gpu::GpusolverRewriter
189 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
Dhorizontal_input_fusion.h53 StatusOr<bool> RunOnComputation(HloComputation*);
Dcusolver_rewriter.h41 StatusOr<bool> RunOnComputation(HloComputation* computation);
Dhorizontal_loop_fusion.h111 StatusOr<bool> RunOnComputation(HloComputation*);
Dgpu_conv_padding_legalization.h39 StatusOr<bool> RunOnComputation(HloComputation* computation);
Dgpu_conv_algorithm_picker.h59 StatusOr<bool> RunOnComputation(HloComputation* computation);
Dgemm_rewriter.cc359 StatusOr<bool> RunOnComputation(HloComputation *computation) { in RunOnComputation() function
373 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
Dgemm_algorithm_picker.cc365 StatusOr<bool> RunOnComputation(HloComputation* computation, in RunOnComputation() function
394 bool result, RunOnComputation(computation, stream_exec_, allocator_)); in Run()
Dgpu_conv_padding_legalization.cc404 StatusOr<bool> GpuConvPaddingLegalization::RunOnComputation( in RunOnComputation() function in xla::gpu::GpuConvPaddingLegalization
436 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
Dhorizontal_loop_fusion.cc535 StatusOr<bool> GpuHorizontalLoopFusion::RunOnComputation( in RunOnComputation() function in xla::gpu::GpuHorizontalLoopFusion
548 TF_ASSIGN_OR_RETURN(changed, RunOnComputation(module->entry_computation())); in Run()
Dgpu_conv_rewriter.cc729 StatusOr<bool> RunOnComputation(HloComputation* computation) { in RunOnComputation() function
753 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
Dgpu_conv_algorithm_picker.cc982 StatusOr<bool> GpuConvAlgorithmPicker::RunOnComputation( in RunOnComputation() function in xla::gpu::GpuConvAlgorithmPicker
1013 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dcanonicalize_all_gather_for_cse.cc27 StatusOr<bool> CanonicalizeAllGatherForCSE::RunOnComputation( in RunOnComputation() function in xla::CanonicalizeAllGatherForCSE
101 TF_ASSIGN_OR_RETURN(bool comp_changed, RunOnComputation(comp)); in Run()
Dschedule_aware_collective_ops_cse.cc88 StatusOr<bool> RunOnComputation(HloComputation* comp, bool for_replicas, in RunOnComputation() function
166 RunOnComputation(comp, for_replicas_, distance_threshold_)); in Run()
Dcanonicalize_all_gather_for_cse.h38 StatusOr<bool> RunOnComputation(HloComputation* comp);