Searched refs:alias_ (Results 1 – 4 of 4) sorted by relevance
24 return alias_.element(output_index).has_value(); in OutputHasAlias()33 TF_RET_CHECK(ShapeUtil::IndexIsValid(alias_.shape(), output_index)) in SetUpAlias()36 << ShapeUtil::HumanString(alias_.shape()); in SetUpAlias()41 TF_RET_CHECK(!alias_.element(output_index)) << absl::StrFormat( in SetUpAlias()45 alias_.element(output_index)->parameter_number, in SetUpAlias()46 alias_.element(output_index)->parameter_index.ToString()); in SetUpAlias()47 (*alias_.mutable_element(output_index)) = in SetUpAlias()57 alias_.ForEachElement( in ToProto()106 const Shape& HloInputOutputAliasConfig::shape() const { return alias_.shape(); } in shape()112 absl::StrFormat(" Output shape: %s", alias_.shape().ToString())); in ToString()[all …]
60 : alias_(std::move(output_shape)) {} in HloInputOutputAliasConfig()131 ShapeTree<absl::optional<Alias>> alias_;
490 alias_(alias) {} in ClassType()509 CurrentScope::Scope scope_activator(alias_->ParentScope()); in Finalize()
768 const TypeAlias* alias_; variable