Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc225 const xla::HloInputOutputAliasConfig& input_output_alias) { in PopulateInputs()
303 const xla::HloInputOutputAliasConfig& input_output_alias, in GetOrCreateTensorForOutput()
425 const xla::HloInputOutputAliasConfig& input_output_alias, in PopulateOutputs()
Dxla_compile_on_demand_op.cc69 const xla::HloInputOutputAliasConfig& input_output_alias = in Run() local
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc301 const xla::HloInputOutputAliasConfig& input_output_alias) { in RebuildOutputAliases()
313 const xla::HloInputOutputAliasConfig& input_output_alias, in GetArgumentsBuffers()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc261 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local
483 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc140 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTuple()
176 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTensors()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dpromote_resources_to_args.cc280 llvm::SmallVector<std::pair<int64_t, int64_t>, 4> input_output_alias; in PromoteResourcesToArguments() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc233 const HloInputOutputAliasConfig& input_output_alias = in CreateResultShapedBuffer() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo.proto398 HloInputOutputAliasProto input_output_alias = 8; field