| /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
166 RebuildOutputAliases(output_tuple_ptr, input_tuples, input_output_alias)); in AllocateOutputTuple()
177 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTensors() argument
182 input_output_alias, std::move(output_scoped_buffer), in AllocateOutputTensors()
337 xla::HloInputOutputAliasConfig input_output_alias, in DoWork()
351 return GetArgumentsBuffers(input_output_alias, input_tuples, in DoWork()
368 executable, input_tuples, input_output_alias, output.ConsumeResult(), in DoWork()
448 xla::HloInputOutputAliasConfig input_output_alias, in DoWork()
453 return GetArgumentsBuffers(input_output_alias, input_tuples, in DoWork()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 238 const xla::HloInputOutputAliasConfig& input_output_alias) { in PopulateInputs() argument
266 input_output_alias.ParameterHasAlias(i, xla::ShapeIndex{}); in PopulateInputs()
319 const xla::HloInputOutputAliasConfig& input_output_alias, in GetOrCreateTensorForOutput() argument
325 xla::ShapeIndex output_index = input_output_alias.shape().IsTuple() in GetOrCreateTensorForOutput()
328 CHECK(input_output_alias.shape().IsTuple() || output_num == 0); in GetOrCreateTensorForOutput()
330 input_output_alias.GetAliasedParameter(output_index)) { in GetOrCreateTensorForOutput()
457 const xla::HloInputOutputAliasConfig& input_output_alias, in PopulateOutputs() argument
556 input_output_alias, compilation_result->input_mapping, in PopulateOutputs()
594 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()
102 input_output_alias, snapshot_ptrs)); in Run()
|
| D | xla_launch_util.h | 167 const xla::HloInputOutputAliasConfig& input_output_alias);
184 const xla::HloInputOutputAliasConfig& input_output_alias,
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.h | 87 const xla::HloInputOutputAliasConfig& input_output_alias);
90 const xla::HloInputOutputAliasConfig& input_output_alias,
|
| D | xrt_util.cc | 303 const xla::HloInputOutputAliasConfig& input_output_alias) { in RebuildOutputAliases() argument
311 return input_output_alias.ForEachAliasWithStatus(alias_function); in RebuildOutputAliases()
315 const xla::HloInputOutputAliasConfig& input_output_alias, in GetArgumentsBuffers() argument
334 if (input_output_alias.ParameterHasAlias(i, index)) { in GetArgumentsBuffers()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 266 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local
271 input_output_alias); in Compute()
314 input_output_alias, resource_var_ptrs)); in Compute()
509 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute() local
529 input_output_alias); in Compute()
570 absl::MakeSpan(*variable_infos), input_output_alias, snapshot_ptrs)); in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | promote_resources_to_args.cc | 260 llvm::SmallVector<std::pair<int64_t, int64_t>, 4> input_output_alias; in PromoteResourcesToArguments() local
284 input_output_alias.push_back( in PromoteResourcesToArguments()
334 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 | 236 const HloInputOutputAliasConfig& input_output_alias = in CreateResultShapedBuffer() local
263 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 | 414 entry->ComputeProgramShape().result(), proto.input_output_alias())); in CreateFromProto()
|
| D | hlo.proto | 443 HloInputOutputAliasProto input_output_alias = 8; field
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 1906 result.computation->proto().input_output_alias(); in TEST_F()
|