Searched defs:alias_config (Results 1 – 7 of 7) sorted by relevance
39 const auto& alias_config = module->input_output_alias_config(); in TEST_F() local55 const auto& alias_config = module->input_output_alias_config(); in TEST_F() local
67 absl::optional<HloInputOutputAliasConfig::Alias> alias_config; member
42 HloInputOutputAliasConfig* alias_config) { in Build()96 HloInputOutputAliasConfig* alias_config = in Run() local
123 HloInputOutputAliasConfig alias_config, int device_ordinal, in ExecutionInputsFromScopedShapedBuffers()
818 HloInputOutputAliasConfig alias_config(module->result_shape()); in ParseHloModule() local
55 const Shape& host_shape, const HloInputOutputAliasConfig& alias_config, in AllocateOutputMemoryWithInputReuse()182 const HloInputOutputAliasConfig& alias_config = in ExecuteAsyncOnStream() local
72 HloInputOutputAliasConfig alias_config = in RunAndCheck() local