Home
last modified time | relevance | path

Searched refs:CreateVariantFromRagged (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_variant_op_test.cc66 RaggedTensorVariant CreateVariantFromRagged( in CreateVariantFromRagged() function in tensorflow::__anon3ae4a31b0111::RaggedTensorToVariantKernelTest
85 RaggedTensorVariant CreateVariantFromRagged( in CreateVariantFromRagged() function in tensorflow::__anon3ae4a31b0111::RaggedTensorToVariantKernelTest
89 return CreateVariantFromRagged(ragged_splits, {num_values}, ragged_values); in CreateVariantFromRagged()
116 CreateVariantFromRagged<int, int64>({{0, 0, 0}}, {}), in TEST_F()
119 CreateVariantFromRagged<int, int64>({{0, 0}}, {}), in TEST_F()
122 CreateVariantFromRagged<int, int64>({{0}}, {}), in TEST_F()
143 CreateVariantFromRagged<int, int64>({}, {1, 2, 3}),
146 CreateVariantFromRagged<int, int64>({}, {}),
149 CreateVariantFromRagged<int, int64>({}, {4, 5}),
152 CreateVariantFromRagged<int, int64>({}, {6}),
[all …]
Dragged_tensor_from_variant_op_test.cc59 RaggedTensorVariant CreateVariantFromRagged( in CreateVariantFromRagged() function in tensorflow::__anon9c8e6f2c0111::RaggedTensorFromVariantKernelTest
83 auto encoded_variant = CreateVariantFromRagged<int, int64>( in TEST_F()
111 auto encoded_variant = CreateVariantFromRagged<int, int64>( in TEST_F()
156 CreateVariantFromRagged<int, int64>({}, TensorShape({3}), values_1); in TEST_F()
158 CreateVariantFromRagged<int, int64>({}, TensorShape({0}), values_2); in TEST_F()
160 CreateVariantFromRagged<int, int64>({}, TensorShape({2}), values_3); in TEST_F()
162 CreateVariantFromRagged<int, int64>({}, TensorShape({1}), values_4); in TEST_F()
221 auto variant_component_1 = CreateVariantFromRagged<int, int64>( in TEST_F()
223 auto variant_component_2 = CreateVariantFromRagged<int, int64>( in TEST_F()
225 auto variant_component_3 = CreateVariantFromRagged<int, int64>( in TEST_F()
[all …]