Searched defs:alias_config (Results 1 – 8 of 8) sorted by relevance
40 const auto& alias_config = module->input_output_alias_config(); in TEST_F() local56 const auto& alias_config = module->input_output_alias_config(); in TEST_F() local
102 std::optional<HloInputOutputAliasConfig::Alias> alias_config; member
42 HloInputOutputAliasConfig* alias_config) { in Build()98 HloInputOutputAliasConfig* alias_config = in Run() local
131 HloInputOutputAliasConfig alias_config, int device_ordinal, in ExecutionInputsFromScopedShapedBuffers()
961 HloInputOutputAliasConfig alias_config(module->result_shape()); in ParseHloModule() local
55 const Shape& shape, const HloInputOutputAliasConfig& alias_config, in AllocateOutputMemoryWithInputReuse()197 const HloInputOutputAliasConfig& alias_config = in ExecuteAsyncOnStream() local
131 const HloInputOutputAliasConfig& alias_config = in ExecuteAsyncOnStream() local155 const Shape& shape, const HloInputOutputAliasConfig& alias_config, in AllocateOutputMemoryWithInputReuse()
74 HloInputOutputAliasConfig alias_config = in RunAndCheck() local