| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 134 .input_output_alias()); in GetExecutableAliasConfig()
140 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTuple() argument
165 RebuildOutputAliases(output_tuple_ptr, input_tuples, input_output_alias)); in AllocateOutputTuple()
176 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTensors() argument
181 input_output_alias, std::move(output_scoped_buffer), in AllocateOutputTensors()
335 xla::HloInputOutputAliasConfig input_output_alias, in DoWork()
349 return GetArgumentsBuffers(input_output_alias, input_tuples, in DoWork()
366 executable, input_tuples, input_output_alias, output.ConsumeResult(), in DoWork()
446 xla::HloInputOutputAliasConfig input_output_alias, in DoWork()
451 return GetArgumentsBuffers(input_output_alias, input_tuples, in DoWork()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 225 const xla::HloInputOutputAliasConfig& input_output_alias) { in PopulateInputs() argument
253 input_output_alias.ParameterHasAlias(i, xla::ShapeIndex{}); in PopulateInputs()
303 const xla::HloInputOutputAliasConfig& input_output_alias, in GetOrCreateTensorForOutput() argument
308 xla::ShapeIndex output_index = input_output_alias.shape().IsTuple() in GetOrCreateTensorForOutput()
312 CHECK(input_output_alias.shape().IsTuple() || output_num == 0); in GetOrCreateTensorForOutput()
314 input_output_alias.GetAliasedParameter(output_index)) { in GetOrCreateTensorForOutput()
425 const xla::HloInputOutputAliasConfig& input_output_alias, in PopulateOutputs() argument
519 output_num, ctx, missing_ctx_input_prefix, input_output_alias, in PopulateOutputs()
566 output_num, ctx, missing_ctx_input_prefix, input_output_alias, in PopulateOutputs()
|
| D | xla_compile_on_demand_op.cc | 69 const xla::HloInputOutputAliasConfig& input_output_alias = in Run() local
74 input_output_alias); in Run()
98 input_output_alias, snapshot_ptrs)); in Run()
|
| D | xla_launch_util.h | 161 const xla::HloInputOutputAliasConfig& input_output_alias);
178 const xla::HloInputOutputAliasConfig& input_output_alias,
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.h | 86 const xla::HloInputOutputAliasConfig& input_output_alias);
89 const xla::HloInputOutputAliasConfig& input_output_alias,
|
| D | xrt_util.cc | 301 const xla::HloInputOutputAliasConfig& input_output_alias) { in RebuildOutputAliases() argument
309 return input_output_alias.ForEachAliasWithStatus(alias_function); in RebuildOutputAliases()
313 const xla::HloInputOutputAliasConfig& input_output_alias, in GetArgumentsBuffers() argument
332 if (input_output_alias.ParameterHasAlias(i, index)) { in GetArgumentsBuffers()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 261 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local
266 input_output_alias); in Compute()
295 input_output_alias, resource_var_ptrs)); in Compute()
483 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local
503 input_output_alias); in Compute()
544 absl::MakeSpan(*variable_infos), input_output_alias, snapshot_ptrs)); in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | promote_resources_to_args.cc | 280 llvm::SmallVector<std::pair<int64_t, int64_t>, 4> input_output_alias; in PromoteResourcesToArguments() local
304 input_output_alias.push_back( in PromoteResourcesToArguments()
354 for (auto& input_output : input_output_alias) in PromoteResourcesToArguments()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/ |
| D | graph-resource.mlir | 16 // CHECK-LABEL: HloModule main.{{[0-9]+}}, input_output_alias={ {0}: (1, {}, may-alias) }
|
| D | graph-resource.pbtxt | 55 # CHECK-LABEL: HloModule main.{{[0-9]+}}, input_output_alias={ {0}: (1, {}, may-alias) }
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 233 const HloInputOutputAliasConfig& input_output_alias = in CreateResultShapedBuffer() local
260 input_output_alias.GetAliasedParameter(index); in CreateResultShapedBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | aliasing.md | 30 HloModule increment, input_output_alias={ {}: 0 }
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 469 hlo_metadata.hlo_module().input_output_alias())); in TPUExecute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module.cc | 407 entry->ComputeProgramShape().result(), proto.input_output_alias())); in CreateFromProto()
|
| D | hlo.proto | 398 HloInputOutputAliasProto input_output_alias = 8; field
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 1895 result.computation->proto().input_output_alias(); in TEST_F()
|