| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_liveness_analysis.cc | 285 for (auto* computation : module_.computations()) { in RunAnalysis() local
|
| H A D | all_reduce_combiner_test.cc | 120 auto computation = module->AddEntryComputation(b.Build()); in TEST_F() local
|
| H A D | hlo_module.h | 415 std::unique_ptr<HloComputation> computation, bool is_entry) { in AddComputationAndUnifyNamesAndIds()
|
| H A D | hlo_instructions.h | 1758 void set_select(HloComputation* computation) { in set_select()
1765 void set_scatter(HloComputation* computation) { in set_scatter()
|
| H A D | algebraic_simplifier.cc | 412 void AlgebraicSimplifierVisitor::ResetState(HloComputation* computation) { in ResetState()
417 bool AlgebraicSimplifierVisitor::Run(HloComputation* computation, in Run()
1570 static HloInstruction* BuildTupleConstant(HloComputation* computation, in BuildTupleConstant()
1677 HloInstruction* divide, HloComputation* computation, in TryDivideToShift()
4061 HloInstruction* remainder, HloComputation* computation, in TryRemainderToAnd()
|
| H A D | hlo_evaluator.cc | 772 const HloComputation& computation, in Evaluate()
3122 auto* computation = call->to_apply(); in HandleCall() local
3585 static bool IsScalarAdd(HloComputation* computation) { in IsScalarAdd()
3607 HloComputation* computation, HloEvaluator* embedded_evaluator) { in PerformReductionStep()
|
| H A D | hlo_replication_analysis.cc | 228 const HloComputation* computation, bool mark_everything_not_replicated) { in ComputeHloReplicationOnComputation()
|
| H A D | hlo_domain_test.cc | 729 auto computation = tuple0->parent(); in TEST_F() local
|
| H A D | qr_expander.cc | 520 HloComputation*& computation = in ExpandInstruction() local
|
| H A D | dfs_hlo_visitor_with_default.h | 300 for (const auto& computation : variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 129 void IrEmitter::EmitThreadLocalFunctionEpilogue(HloComputation* computation) { in EmitThreadLocalFunctionEpilogue()
167 HloComputation* computation, const std::string& function_name_prefix, in EmitComputation()
2230 HloComputation* computation = call->to_apply(); in HandleCall() local
3010 const HloComputation& computation) { in GetProfileCounterFor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| H A D | parallel_device_test.py | 467 def _assert_close_to_non_parallel(self, computation): argument
|
| /aosp_15_r20/external/federated-compute/fcp/client/ |
| H A D | lc_runner.cc | 280 LocalComputation computation = LocalComputation(); in RunLocalComputation() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/python/ |
| H A D | init.cpp | 312 auto computation = LazyGraphExecutor::Get() in initLazyBindings() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | outfeed_receiver.cc | 404 XlaComputation computation = builder.Build(send).ValueOrDie(); in SendShutdownOutfeedHeader() local
|
| H A D | outfeed_receiver_test.cc | 36 XlaComputation computation = builder->Build(root).ValueOrDie(); in CompileAndExecute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | xla_hlo_profile_test.cc | 132 const XlaComputation& computation, in ExecuteAndFetchProfile()
|
| H A D | reshape_test.cc | 649 XlaComputation computation = builder.Build().value(); in XLA_TEST_P() local
804 XlaComputation computation = builder.Build().value(); in XLA_TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | pjrt_c_api_client.h | 162 const XlaComputation& computation, CompileOptions options) override { in Compile()
|
| H A D | tfrt_cpu_pjrt_client.cc | 247 const XlaComputation& computation, in JitCompile()
333 const XlaComputation& computation, CompileOptions options) { in Compile()
|
| H A D | pjrt_client_test.cc | 85 XlaComputation computation = builder.Build(inc).value(); in MakeIncrementProgram() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | mhlo_to_lhlo_with_xla.cc | 1361 Status LhloDialectEmitter::ImportAsLmhloRegion(xla::HloComputation* computation, in ImportAsLmhloRegion()
1723 const HloComputation* computation = hlo_module.entry_computation(); in HloToLhloModule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_support.cc | 343 const absl::variant<MlirToHloArgs, FunctionToHloArgs>& computation, in CreateTpuCompilationRequest()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | nccl_all_reduce_thunk.cc | 174 mlir::Region& computation) { in MatchAllReduceComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | value_inference.cc | 183 auto computation = builder.Build(); in Evaluate() local
201 HloComputation* computation = nullptr; member
|