Searched refs:inference_attrs_ (Results 1 – 2 of 2) sorted by relevance
369 inference_attrs_.find(type_attr) == inference_attrs_.end()) { in MaybeInferSingleInputAttrs()371 inference_attrs_.insert(type_attr); in MaybeInferSingleInputAttrs()378 if (inference_attrs_.find(input_def.number_attr()) == in InferSingleTypeInputListAttrs()379 inference_attrs_.end()) { in InferSingleTypeInputListAttrs()381 inference_attrs_.insert(input_def.number_attr()); in InferSingleTypeInputListAttrs()383 if (inference_attrs_.find(input_def.type_attr()) == inference_attrs_.end()) { in InferSingleTypeInputListAttrs()385 inference_attrs_.insert(input_def.type_attr()); in InferSingleTypeInputListAttrs()391 if (inference_attrs_.find(input_def.type_list_attr()) == in InferMixedTypeInputListAttrs()392 inference_attrs_.end()) { in InferMixedTypeInputListAttrs()396 inference_attrs_.insert(input_def.type_list_attr()); in InferMixedTypeInputListAttrs()[all …]
208 inference_attrs_.clear_no_resize(); in ClearInferenceState()261 gtl::FlatSet<std::string> inference_attrs_; // attributes inferred so far variable