Home
last modified time | relevance | path

Searched refs:MultiaryUnstack (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfull_type_inference_util_test.cc380 TEST(MultiaryUnstack, Basic) { in TEST() argument
384 const auto ret = MultiaryUnstack(TFT_DATASET, UnstackTensor)({t1}, {}); in TEST()
397 TEST(MultiaryUnstack, Ternary) { in TEST() argument
408 MultiaryUnstack(TFT_DATASET, UnstackTensor)({t1, t2, t3}, {}); in TEST()
H A Dfull_type_inference_util.h97 ForwardTypeInferenceFn MultiaryUnstack(
H A Dfull_type_inference_util.cc228 ForwardTypeInferenceFn MultiaryUnstack( in MultiaryUnstack() function
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Ddataset_ops.cc63 .SetForwardTypeFn(full_type::MultiaryUnstack(TFT_DATASET,