| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_cost_analysis_test.cc | 119 auto computation = std::move(computation_status).value(); in BuildHloGraph() local
727 auto* computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
789 auto* computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
843 auto* computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
|
| H A D | scatter_expander.cc | 138 HloComputation* computation = index_vector->parent(); in ExpandIndexVectorIntoOperandSpace() local
175 HloComputation* computation, HloInstruction* index, in CheckIndexValidity()
|
| H A D | conditional_code_motion.cc | 441 Status RestructureConditionalInstruction(HloComputation* computation, in RestructureConditionalInstruction()
683 auto computation = conditional->branch_computation(i); in MoveInstructionOut() local
763 auto computation = conditional->branch_computation(i); in MoveInstructionIn() local
842 auto computation = conditional->branch_computation(i); in MoveInstructionIn() local
|
| H A D | hlo_instruction.h | 1832 void set_parent(HloComputation* computation) { parent_ = computation; } in set_parent()
2185 void AppendComputation(HloComputation* computation) { in AppendComputation()
2191 void set_called_computation(int index, HloComputation* computation) { in set_called_computation()
|
| H A D | memory_space_assignment.cc | 358 const HloComputation* computation = instruction->parent(); in CalculateComputationNestLevel() local
2450 for (const HloComputation* computation : module.MakeNonfusionComputations()) { in AddInputAndOutputRequiredAssignments() local
3405 for (const HloComputation* computation : in CalculateAsyncCopyStats() local
3602 HloComputation* computation = producing_instruction->parent(); in Process() local
3676 HloComputation* computation = producing_instruction->parent(); in Process() local
4094 for (const HloComputation* computation : in FixSchedule() local
4215 for (const HloComputation* computation : in VerifyAndExportHeapSimulatorTrace() local
|
| H A D | convolution_group_converter.cc | 69 HloComputation* computation, in ConvolutionVisitor()
96 HloComputation* computation, in Run()
|
| H A D | copy_insertion.cc | 54 const HloComputation* computation = value.defining_instruction()->parent(); in IsReadonlyEntryParameterValue() local
82 HloComputation* computation) { in GetSpecialCaseCopyPolicy()
494 InstructionMap& operator[](const HloComputation* computation) { in operator []()
520 auto* computation = instr->parent(); in contains() local
|
| H A D | call_graph_test.cc | 99 HloComputation* computation = in TEST_F() local
508 HloComputation* computation = in TEST_F() local
|
| H A D | batchnorm_expander.cc | 63 explicit BatchNormExpanderVisitor(HloComputation* computation, in BatchNormExpanderVisitor()
138 bool BatchNormExpanderVisitor::Run(HloComputation* computation, in Run()
|
| H A D | convolution_pred_expander.cc | 41 HloComputation* computation = instruction->parent(); in ExpandInstruction() local
|
| H A D | simplify_fp_conversions.cc | 28 StatusOr<bool> RunOnComputation(HloComputation& computation) { in RunOnComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| H A D | tpu_client.cc | 784 for (int computation = 0; computation < num_computations; ++computation) { in ExecuteShardedOnLocalDevices() local
796 for (int computation = 0; computation < num_computations; ++computation) { in ExecuteShardedOnLocalDevices() local
805 const XlaComputation& computation, in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | multioutput_fusion_test.cc | 92 auto computation = hlo_module->AddEntryComputation(builder.Build(dot)); in RunTest2D() local
155 auto computation = hlo_module->AddEntryComputation(builder.Build(dot)); in RunTest1D() local
|
| H A D | reduce_window_test.cc | 649 auto computation = reducer == kAdd in DoIt() local
1077 auto computation = reducer == kAdd in XLA_TEST_P() local
1308 auto computation = param.reducer == kAdd in DoIt() local
1533 auto computation = param.reducer == kAdd in XLA_TEST_P() local
|
| /aosp_15_r20/external/tink/cc/mac/ |
| H A D | hmac_key_manager_test.cc | 330 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST() local
366 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | local_client.cc | 388 const XlaComputation& computation, in Compile()
411 const XlaComputation& computation, in CompileAheadOfTime()
|
| H A D | compile_only_client.h | 42 const XlaComputation* computation; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | horizontal_loop_fusion.cc | 53 explicit HorizontalLoopFusionImpl(HloComputation* computation, in HorizontalLoopFusionImpl()
536 HloComputation* computation) { in RunOnComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | replay_computation.cc | 115 XlaComputation computation(module.hlo().hlo_module()); in CompileExecutable() local
233 XlaComputation computation(module.hlo().hlo_module()); in ReplayComputation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | pmap_lib.cc | 484 for (int computation = 0; computation < num_computations; ++computation) { in Call() local
495 for (int computation = 0; computation < num_computations; ++computation) { in Call() local
532 for (int computation = 0; computation < num_computations; ++computation) { in Call() local
|
| /aosp_15_r20/development/tools/winscope/src/parsers/view_capture/computations/ |
| H A D | rects_computation_test.ts | 26 let computation: RectsComputation; variable
|
| H A D | visibility_computation_test.ts | 22 let computation: VisibilityComputation; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | hlo_module_importer.cc | 54 for (const auto* computation : module.computations()) in Import() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | gpu_copy_test.cc | 45 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_F() local
|
| /aosp_15_r20/development/tools/winscope/src/parsers/window_manager/computations/ |
| H A D | rects_computation_test.ts | 36 let computation: RectsComputation; variable
|