| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | logical_buffer_analysis.cc | 65 for (auto* computation : module_->MakeNonfusionComputations()) { in Analyze() local
|
| H A D | cholesky_expander.cc | 229 HloComputation*& computation = in ExpandInstruction() local
|
| H A D | dynamic_dimension_inference.cc | 84 static Status Run(HloComputation* computation, in Run()
682 HloComputation* computation = gds->parent(); in HandleGetDimensionSize() local
1820 HloComputation* computation = hlo->parent(); in HandleParameter() local
|
| H A D | collective_ops_utils.cc | 54 const HloComputation* computation) { in MatchReductionComputation()
|
| H A D | hlo_evaluator.h | 134 StatusOr<Literal> Evaluate(const HloComputation& computation, in Evaluate()
|
| H A D | multi_output_fusion.cc | 35 for (auto* computation : in Run() local
|
| H A D | while_util.cc | 288 HloComputation* computation, int32_t trip_count, in MakeCountedLoop()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_nested.cc | 196 Status IrEmitterNested::EmitConstants(const HloComputation& computation) { in EmitConstants()
|
| H A D | multi_output_fusion.cc | 382 for (auto* computation : in Run() local
|
| H A D | hlo_to_ir_bindings.cc | 214 for (const HloComputation* computation : in ToString() local
|
| H A D | gpu_layout_assignment_test.cc | 70 HloComputation* computation = in TEST_F() local
|
| H A D | gemm_rewriter.cc | 359 StatusOr<bool> RunOnComputation(HloComputation *computation) { in RunOnComputation()
|
| H A D | gemm_algorithm_picker.cc | 365 StatusOr<bool> RunOnComputation(HloComputation* computation, in RunOnComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executable_base.cc | 77 const HloComputation* computation = module().entry_computation(); in ExecuteAsyncOnStream() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | xrt_compile_ops.cc | 110 Status CompilationCacheKey(const xrt::XLAComputation& computation, in CompilationCacheKey()
|
| H A D | tpu_compile_ops.cc | 92 const xrt::XLAComputation& computation, in CompilationCacheKey()
|
| /aosp_15_r20/external/perfetto/src/trace_processor/sqlite/ |
| H A D | db_sqlite_table.h | 55 TableComputation computation; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | tpu_client.cc | 180 XlaComputation computation(tpu_executable->module().ToProto()); in DeserializeExecutable() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | deconstruct_tuple_test.cc | 47 XlaComputation computation = builder->Build().value(); in ExecuteAndCheckTransfer() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/xla/ |
| H A D | xla_test.py | 221 def computation(a, b): function
|
| /aosp_15_r20/external/tink/cc/mac/ |
| H A D | mac_config_test.cc | 366 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST_P() local
|
| /aosp_15_r20/external/tink/cc/mac/internal/ |
| H A D | chunked_mac_wrapper_test.cc | 146 util::StatusOr<std::unique_ptr<ChunkedMacComputation>> computation = in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| H A D | tpu_client_extension.cc | 143 -> StatusOr<std::unique_ptr<PyTpuExecutable>> { in PYBIND11_MODULE()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_backend_impl.cpp | 208 torch::lazy::ComputationPtr computation, in ExecuteComputation()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | ir_dump_util.cpp | 254 auto computation = lowering_ctx->Build(); in ToBackend() local
|