Home
last modified time | relevance | path

Searched defs:input_output_alias (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_launch_util.cc252 const xla::HloInputOutputAliasConfig& input_output_alias) { in PopulateInputs()
323 const xla::HloInputOutputAliasConfig& input_output_alias, in GetOrCreateTensorForOutput()
461 const xla::HloInputOutputAliasConfig& input_output_alias, in PopulateOutputs()
H A Dxla_compile_on_demand_op.cc72 const xla::HloInputOutputAliasConfig& input_output_alias = in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/
H A Dxrt_util.cc304 const xla::HloInputOutputAliasConfig& input_output_alias) { in RebuildOutputAliases()
316 const xla::HloInputOutputAliasConfig& input_output_alias, in GetArgumentsBuffers()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dtpu_execute_op.cc140 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTuple()
177 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTensors()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/
H A Dxla_ops.cc346 const xla::HloInputOutputAliasConfig& input_output_alias = in ComputeAsync() local
589 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_executable.cc226 const HloInputOutputAliasConfig& input_output_alias = in CreateResultShapedBuffer() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dpromote_resources_to_args.cc263 llvm::SmallVector<std::pair<int64_t, int64_t>, 4> input_output_alias; in PromoteResourcesToArguments() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo.proto481 HloInputOutputAliasProto input_output_alias = 8; field