Searched defs:input_output_alias (Results 1 – 8 of 8) sorted by relevance
238 const xla::HloInputOutputAliasConfig& input_output_alias) { in PopulateInputs()319 const xla::HloInputOutputAliasConfig& input_output_alias, in GetOrCreateTensorForOutput()457 const xla::HloInputOutputAliasConfig& input_output_alias, in PopulateOutputs()
69 const xla::HloInputOutputAliasConfig& input_output_alias = in Run() local
303 const xla::HloInputOutputAliasConfig& input_output_alias) { in RebuildOutputAliases()315 const xla::HloInputOutputAliasConfig& input_output_alias, in GetArgumentsBuffers()
140 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTuple()177 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTensors()
266 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local509 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local
236 const HloInputOutputAliasConfig& input_output_alias = in CreateResultShapedBuffer() local
260 llvm::SmallVector<std::pair<int64_t, int64_t>, 4> input_output_alias; in PromoteResourcesToArguments() local
443 HloInputOutputAliasProto input_output_alias = 8; field