| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dce.cc | 40 StatusOr<bool> HloDCE::RunOnComputation( in RunOnComputation() function in xla::HloDCE
86 RunOnComputation(computation, remove_cross_partition_collective_ops_)); in Run()
|
| D | hlo_dce.h | 46 StatusOr<bool> RunOnComputation(HloComputation* computation,
|
| D | hlo_module_dce.cc | 97 HloDCE().RunOnComputation( in RunWhileDCE()
|
| D | layout_assignment.h | 407 Status RunOnComputation(ComputationLayout* computation_layout,
|
| D | layout_assignment.cc | 1841 Status LayoutAssignment::RunOnComputation( in RunOnComputation() function in xla::LayoutAssignment
2179 TF_RETURN_IF_ERROR(RunOnComputation(entry_computation_layout_, in Run()
2187 TF_RETURN_IF_ERROR(RunOnComputation(computation_layout, computation, in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | schedule_aware_all_gather_cse.cc | 54 StatusOr<bool> RunOnComputation(HloComputation* comp, bool for_replicas, in RunOnComputation() function
120 RunOnComputation(comp, for_replicas_, distance_threshold_)); in Run()
|
| D | canonicalize_all_gather_for_cse.cc | 44 StatusOr<bool> CanonicalizeAllGatherForCSE::RunOnComputation( in RunOnComputation() function in xla::CanonicalizeAllGatherForCSE
101 TF_ASSIGN_OR_RETURN(bool comp_changed, RunOnComputation(comp)); in Run()
|
| D | canonicalize_all_gather_for_cse.h | 35 StatusOr<bool> RunOnComputation(HloComputation* comp);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_input_fusion.cc | 150 StatusOr<bool> GpuHorizontalInputFusion::RunOnComputation( in RunOnComputation() function in xla::gpu::GpuHorizontalInputFusion
160 TF_ASSIGN_OR_RETURN(changed, RunOnComputation(comp)); in Run()
|
| D | gemm_rewriter.cc | 127 static StatusOr<bool> RunOnComputation(HloComputation *computation) { in RunOnComputation() function
136 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
|
| D | horizontal_input_fusion.h | 51 StatusOr<bool> RunOnComputation(HloComputation*);
|
| D | horizontal_loop_fusion.h | 108 StatusOr<bool> RunOnComputation(HloComputation*);
|
| D | cusolver_rewriter.h | 38 StatusOr<bool> RunOnComputation(HloComputation* computation);
|
| D | gpu_conv_padding_legalization.h | 36 StatusOr<bool> RunOnComputation(HloComputation* computation);
|
| D | cusolver_rewriter.cc | 153 StatusOr<bool> CusolverRewriter::RunOnComputation(HloComputation* computation) { in RunOnComputation() function in xla::gpu::CusolverRewriter
181 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
|
| D | gpu_conv_algorithm_picker.h | 51 StatusOr<bool> RunOnComputation(HloComputation* computation);
|
| D | gemm_algorithm_picker.cc | 302 static StatusOr<bool> RunOnComputation(HloComputation* computation, in RunOnComputation() function
326 bool result, RunOnComputation(computation, stream_exec_, allocator_)); in Run()
|
| D | horizontal_loop_fusion.cc | 497 StatusOr<bool> GpuHorizontalLoopFusion::RunOnComputation( in RunOnComputation() function in xla::gpu::GpuHorizontalLoopFusion
507 TF_ASSIGN_OR_RETURN(changed, RunOnComputation(comp)); in Run()
|
| D | gpu_conv_padding_legalization.cc | 400 StatusOr<bool> GpuConvPaddingLegalization::RunOnComputation( in RunOnComputation() function in xla::gpu::GpuConvPaddingLegalization
429 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
|
| D | gpu_conv_rewriter.cc | 718 StatusOr<bool> RunOnComputation(HloComputation* computation) { in RunOnComputation() function
739 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
|
| D | gpu_conv_algorithm_picker.cc | 819 StatusOr<bool> GpuConvAlgorithmPicker::RunOnComputation( in RunOnComputation() function in xla::gpu::GpuConvAlgorithmPicker
847 TF_ASSIGN_OR_RETURN(bool result, RunOnComputation(computation)); in Run()
|