| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_input_output_alias_config.cc | 23 bool HloInputOutputAliasConfig::OutputHasAlias( in OutputHasAlias()
28 Status HloInputOutputAliasConfig::SetUpAlias( in SetUpAlias()
31 HloInputOutputAliasConfig::AliasKind must_alias) { in SetUpAlias()
54 HloInputOutputAliasProto HloInputOutputAliasConfig::ToProto() const { in ToProto()
78 StatusOr<HloInputOutputAliasConfig> HloInputOutputAliasConfig::CreateFromProto( in CreateFromProto()
80 HloInputOutputAliasConfig result(std::move(output_shape)); in CreateFromProto()
95 const Shape& HloInputOutputAliasConfig::shape() const { return alias_.shape(); } in shape()
97 string HloInputOutputAliasConfig::ToString() const { in ToString()
112 string HloInputOutputAliasConfig::ToShortString() const { in ToShortString()
124 bool HloInputOutputAliasConfig::ParameterMustAlias( in ParameterMustAlias()
[all …]
|
| D | hlo_input_output_alias_config_test.cc | 39 const HloInputOutputAliasConfig& config) { in expect_aliased()
46 absl::optional<HloInputOutputAliasConfig::Alias> aliased_param = in expect_aliased()
56 const HloInputOutputAliasConfig& config) { in expect_not_aliased()
62 absl::optional<HloInputOutputAliasConfig::Alias> aliased_param = in expect_not_aliased()
84 HloInputOutputAliasConfig config( in TEST_F()
115 HloInputOutputAliasConfig config( in TEST_F()
152 HloInputOutputAliasConfig config( in TEST_F()
181 HloInputOutputAliasConfig config( in TEST_F()
206 HloInputOutputAliasConfig config( in TEST_F()
|
| D | hlo_input_output_alias_config.h | 33 class HloInputOutputAliasConfig {
64 HloInputOutputAliasConfig() = default;
66 explicit HloInputOutputAliasConfig(Shape output_shape) in HloInputOutputAliasConfig() function
69 virtual ~HloInputOutputAliasConfig() = default;
91 static StatusOr<HloInputOutputAliasConfig> CreateFromProto(
145 const HloInputOutputAliasConfig& config);
|
| D | optimize_input_output_buffer_alias_test.cc | 48 [&](const ShapeIndex&, const HloInputOutputAliasConfig::Alias&) { in AliasCount()
55 config_ = HloInputOutputAliasConfig(output_shape); in BuildAliasConfig()
64 HloInputOutputAliasConfig config_;
|
| D | optimize_input_output_buffer_alias.cc | 42 HloInputOutputAliasConfig* alias_config) { in Build()
96 HloInputOutputAliasConfig* alias_config = in Run()
|
| D | hlo_module.h | 282 HloInputOutputAliasConfig& input_output_alias_config() { in input_output_alias_config()
285 const HloInputOutputAliasConfig& input_output_alias_config() const { in input_output_alias_config()
417 HloInputOutputAliasConfig input_output_alias_config_;
|
| D | optimize_input_output_buffer_alias.h | 62 HloInputOutputAliasConfig* alias_config);
|
| D | hlo_module.cc | 63 input_output_alias_config_ = HloInputOutputAliasConfig( in ReplaceEntryComputation()
83 input_output_alias_config_ = HloInputOutputAliasConfig( in AddComputationInternal()
406 HloInputOutputAliasConfig::CreateFromProto( in CreateFromProto()
|
| D | copy_insertion.cc | 366 const HloInputOutputAliasConfig::Alias& alias) { in AddCopiesForAliasedInputOutputs()
413 const HloInputOutputAliasConfig::Alias& alias) -> Status { in AddCopiesForAliasedInputOutputs()
1109 absl::optional<HloInputOutputAliasConfig::Alias> alias = in AddSpecialCaseCopies()
|
| D | hlo_runner.cc | 123 HloInputOutputAliasConfig alias_config, int device_ordinal, in ExecutionInputsFromScopedShapedBuffers()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 55 const Shape& host_shape, const HloInputOutputAliasConfig& alias_config, in AllocateOutputMemoryWithInputReuse()
67 absl::optional<HloInputOutputAliasConfig::Alias> alias) { in AllocateOutputMemoryWithInputReuse()
112 absl::optional<HloInputOutputAliasConfig::Alias> alias = in AllocateOutputMemoryWithInputReuse()
182 const HloInputOutputAliasConfig& alias_config = in ExecuteAsyncOnStream()
183 hlo_module_ == nullptr ? HloInputOutputAliasConfig() in ExecuteAsyncOnStream()
|
| D | tpu_executable_interface.h | 67 const Shape& host_shape, const HloInputOutputAliasConfig& alias_config,
|
| /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()
304 const xla::HloInputOutputAliasConfig::Alias& alias) -> Status { in RebuildOutputAliases()
313 const xla::HloInputOutputAliasConfig& input_output_alias, in GetArgumentsBuffers()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 124 xla::StatusOr<xla::HloInputOutputAliasConfig> GetExecutableAliasConfig( in GetExecutableAliasConfig()
129 return xla::HloInputOutputAliasConfig::CreateFromProto( in GetExecutableAliasConfig()
140 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTuple()
176 const xla::HloInputOutputAliasConfig& input_output_alias, in AllocateOutputTensors()
335 xla::HloInputOutputAliasConfig input_output_alias, in DoWork()
446 xla::HloInputOutputAliasConfig input_output_alias, in DoWork()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.h | 161 const xla::HloInputOutputAliasConfig& input_output_alias);
178 const xla::HloInputOutputAliasConfig& input_output_alias,
|
| D | xla_launch_util.cc | 225 const xla::HloInputOutputAliasConfig& input_output_alias) { in PopulateInputs()
303 const xla::HloInputOutputAliasConfig& input_output_alias, in GetOrCreateTensorForOutput()
313 if (absl::optional<xla::HloInputOutputAliasConfig::Alias> alias = in GetOrCreateTensorForOutput()
425 const xla::HloInputOutputAliasConfig& input_output_alias, in PopulateOutputs()
|
| D | xla_compile_on_demand_op.cc | 69 const xla::HloInputOutputAliasConfig& input_output_alias = in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 72 HloInputOutputAliasConfig alias_config = in RunAndCheck()
147 const HloInputOutputAliasConfig::Alias& alias) { in RunAndCheck()
330 {0}, 0, {}, HloInputOutputAliasConfig::kMustAlias)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | utils.cc | 226 const HloInputOutputAliasConfig& config = module.input_output_alias_config(); in GetParametersThatMustBeDonated()
229 const HloInputOutputAliasConfig::Alias& alias) { in GetParametersThatMustBeDonated()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 233 const HloInputOutputAliasConfig& input_output_alias = in CreateResultShapedBuffer()
259 absl::optional<HloInputOutputAliasConfig::Alias> alias = in CreateResultShapedBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 67 absl::optional<HloInputOutputAliasConfig::Alias> alias_config;
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 261 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute()
483 const xla::HloInputOutputAliasConfig& input_output_alias = in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | xla_mlir_translate.cc | 205 const HloInputOutputAliasConfig::Alias& alias) { in MlirHloToHloTextTranslateFunctionImpl()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 466 xla::HloInputOutputAliasConfig::CreateFromProto( in TPUExecute()
|