Home
last modified time | relevance | path

Searched refs:ForEachAlias (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config.cc103 ForEachAlias([&](const ShapeIndex& output_index, const Alias& alias) { in ToString()
158 void HloInputOutputAliasConfig::ForEachAlias(AliasFn fn) const { in ForEachAlias() function in xla::HloInputOutputAliasConfig
Dhlo_input_output_alias_config.h115 void ForEachAlias(AliasFn fn) const;
Doptimize_input_output_buffer_alias_test.cc47 config_.ForEachAlias( in AliasCount()
Dcopy_insertion.cc364 module->input_output_alias_config().ForEachAlias( in AddCopiesForAliasedInputOutputs()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate.cc203 hlo_module->input_output_alias_config().ForEachAlias( in MlirHloToHloTextTranslateFunctionImpl()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc145 alias_config.ForEachAlias( in RunAndCheck()