Searched refs:SetReverseTypeFn (Results 1 – 4 of 4) sorted by relevance
293 OpDefBuilderWrapper& SetReverseTypeFn(int input_number, in SetReverseTypeFn() function295 builder_.SetReverseTypeFn(input_number, std::move(fn)); in SetReverseTypeFn()
232 OpDefBuilder& SetReverseTypeFn(int input_number, ForwardTypeInferenceFn f);
645 OpDefBuilder& OpDefBuilder::SetReverseTypeFn(int input_number, in SetReverseTypeFn() function in tensorflow::OpDefBuilder
909 .SetReverseTypeFn(1, full_type::MapCovariant(TFT_DATASET, TFT_ITERATOR, 0))