| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | all_reduce_reassociate.cc | 58 for (auto computation : module->computations(execution_threads)) { in Run() local
|
| H A D | all_reduce_combiner.cc | 56 HloComputation& computation = *to_combine.back()->parent(); in CombineAllReduces() local
|
| H A D | all_reduce_contiguous.cc | 35 HloComputation& computation = *all_reduce->parent(); // never null in ReplaceWithContiguousAllReduce() local
|
| H A D | hlo_domain_remover.cc | 106 HloComputation* computation = instruction->parent(); in RemoveExitDomains() local
|
| H A D | tpu_computation_placer.cc | 37 StatusOr<int> TpuComputationPlacer::DeviceId(int replica, int computation, in DeviceId()
|
| H A D | bitcast_dtypes_expander.cc | 63 HloComputation*& computation = in ExpandInstruction() local
|
| H A D | hlo_constant_folding.cc | 80 for (auto* computation : in Run() local
|
| H A D | tree_reduction_rewriter.cc | 118 for (const auto &computation : in Run() local
|
| H A D | hlo_module_dce.cc | 45 for (auto* computation : module->computations(execution_threads)) { in RunWhileDCE() local
|
| H A D | reduce_scatter_combiner.cc | 58 HloComputation& computation = *to_combine.back()->parent(); in CombineReduceScatters() local
|
| H A D | all_gather_combiner.cc | 55 HloComputation& computation = *to_combine.back()->parent(); in CombineAllGathers() local
|
| H A D | collectives_schedule_linearizer_test.cc | 39 int64_t CountControlEdges(const HloComputation& computation) { in CountControlEdges()
|
| H A D | compile_only_service.h | 43 HloModuleProto computation; member
|
| H A D | all_gather_broadcast_reorder.cc | 43 for (auto computation : module->computations(execution_threads)) { in Run() local
|
| H A D | collective_decomposer_utils.cc | 37 int64_t shard_dimension, HloComputation *computation, in CreateStartIndicesForCollectiveDecomposition()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gemm_broadcast_folding_rewriter.cc | 102 static StatusOr<bool> RunOnComputation(HloComputation *computation) { in RunOnComputation()
|
| H A D | hlo_fusion_stats.cc | 82 std::set<std::string> GetUniqueOpcodes(HloComputation* computation) { in GetUniqueOpcodes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner.cc | 145 for (auto computation : module.computations()) { in ReportMemoryUsage() local
1989 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in SpmdPartitioningVisitor()
3641 HloComputation* computation = hlo->branch_computation(i); in HandleConditional() local
3653 HloComputation* computation = hlo->branch_computation(i); in HandleConditional() local
4191 HloComputation* computation, const HloSharding& root_sharding, in DoPartition()
4462 HloComputation* computation, const HloSharding& root_sharding, in PartitionComputation()
4471 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in CreateVisitor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | mlir_hlo_to_hlo.cc | 766 xla::XlaComputation computation; in ExportXlaOp() local
794 xla::XlaComputation computation; in ExportXlaOp() local
1118 xla::XlaComputation& computation = in ExportXlaOp() local
1206 xla::XlaComputation computation; in ExportXlaOp() local
2269 xla::XlaComputation computation; in RunOnFunction() local
|
| /aosp_15_r20/development/tools/winscope/src/parsers/surface_flinger/computations/ |
| H A D | rects_computation_test.ts | 37 let computation: RectsComputation; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | local_client_aot_test_helper.cc | 86 xla::XlaComputation computation = builder.Build().value(); in main() local
|
| H A D | local_client_allocation_test.cc | 79 auto computation = builder.Build().value(); in XLA_TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | dumped_computation_to_text.cc | 52 XlaComputation computation = std::move(computation_status).value(); in RealMain() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | mhlo_to_lhlo_with_xla.h | 46 const xla::HloComputation& computation, ModuleOp module) in LhloDialectEmitter()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_jit_compiled_cpu_function.cc | 92 xla::XlaComputation computation; in Compile() local
|