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.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()
Dxla_compile_on_demand_op.cc72 const xla::HloInputOutputAliasConfig& input_output_alias = in Run() local
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc304 const xla::HloInputOutputAliasConfig& input_output_alias) { in RebuildOutputAliases()
316 const xla::HloInputOutputAliasConfig& input_output_alias, in GetArgumentsBuffers()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc140 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTuple()
177 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTensors()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc346 const xla::HloInputOutputAliasConfig& input_output_alias = in ComputeAsync() local
589 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc226 const HloInputOutputAliasConfig& input_output_alias = in CreateResultShapedBuffer() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dpromote_resources_to_args.cc263 llvm::SmallVector<std::pair<int64_t, int64_t>, 4> input_output_alias; in PromoteResourcesToArguments() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo.proto481 HloInputOutputAliasProto input_output_alias = 8; field