Searched refs:VariadicTensorContainer (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
H A D | experimental_dataset_ops.cc | 27 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 42 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 57 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 72 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 90 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 104 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 114 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 128 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 148 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 159 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, [all …]
|
H A D | dataset_ops.cc | 46 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 61 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 92 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 103 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 114 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 125 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 140 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 155 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 174 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, 191 .SetTypeConstructor(full_type::VariadicTensorContainer(TFT_DATASET, [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | full_type_util.h | 72 OpTypeConstructor VariadicTensorContainer(FullTypeId t, const string& var_name);
|
H A D | full_type_util.cc | 115 OpTypeConstructor VariadicTensorContainer(FullTypeId t, in VariadicTensorContainer() function
|
H A D | full_type_util_test.cc | 112 TEST(VariadicTensorContainer, Basic) { in TEST() argument 113 OpTypeConstructor ctor = VariadicTensorContainer(TFT_ARRAY, "T"); in TEST()
|