| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | value_inference_test.cc | 80 auto computation = ConstantR0<int32_t>(&b, 42); in TEST_F() local
91 auto computation = Iota(&b, S32, 2); in TEST_F() local
198 auto computation = Parameter(&b, 0, ShapeUtil::MakeScalarShape(S32), "p0"); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | const_op_test.py | 98 def computation(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | result_caster.cc | 56 auto* computation = instruction->parent(); in ExpandInstruction() local
|
| H A D | broadcast_canonicalizer.cc | 31 for (const auto& computation : in Run() local
|
| H A D | source_map_util.h | 52 const HloComputation& computation = *module.entry_computation(); in InvalidParameterArgument() local
|
| H A D | memory_space_assignment_utils.cc | 94 for (const HloComputation* computation : module.MakeNonfusionComputations()) { in HoistConstantOperations() local
|
| H A D | tuple_simplifier.cc | 67 for (auto* computation : module->computations(execution_threads)) { in Run() local
|
| H A D | conditional_to_select.cc | 42 auto computation = conditional->parent(); in DoConditionalToSelect() local
|
| H A D | operand_upcaster.cc | 64 auto* computation = instruction->parent(); in ExpandInstruction() local
|
| H A D | collective_combiner_utils.h | 46 HloComputation* computation, in CombineInstructionsByKey()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executable.cc | 56 const HloComputation& computation, absl::Span<const Literal> arg_literals) { in Evaluate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | gpu_noalias_test.cc | 47 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | show_signature.cc | 53 auto computation = client->LoadSnapshot(module).value(); in RealMain() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | batch_ops.py | 91 def computation(*computation_args): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_conv_algorithm_picker.cc | 945 HloComputation* computation = instr->parent(); in RunOnInstruction() local
983 HloComputation* computation) { in RunOnComputation()
|
| H A D | gpu_spmd_partitioner.h | 31 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in GpuSpmdPartitioningVisitor()
|
| H A D | gpu_spmd_partitioner.cc | 50 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in CreateVisitor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | stateful_rng_spmd_partitioner.h | 32 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in StatefulRngSpmdPartitioningVisitor()
|
| H A D | stateful_rng_spmd_partitioner.cc | 50 HloComputation* computation, int64_t num_partitions, int64_t num_replicas, in CreateVisitor()
|
| /aosp_15_r20/development/tools/winscope/src/parsers/surface_flinger/computations/ |
| H A D | z_order_paths_computation_test.ts | 23 let computation: ZOrderPathsComputation; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | sample_harness.cc | 57 xla::XlaComputation computation = builder.Build().value(); in main() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | client.h | 100 const XlaComputation& computation; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | pjrt_compiler.cc | 46 CompileOptions options, const XlaComputation& computation, in PjRtCompile()
|
| /aosp_15_r20/frameworks/base/services/core/java/com/android/server/wm/utils/ |
| H A D | RotationCache.java | 37 public RotationCache(RotationDependentComputation<T, R> computation) { in RotationCache()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_impl.cc | 29 const absl::variant<MlirToHloArgs, FunctionToHloArgs>& computation, in Compile()
|