Home
last modified time | relevance | path

Searched refs:KeyValueTensorInitializer (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dlookup_ops_test.py97 lookup_ops.KeyValueTensorInitializer(keys, values),
125 lookup_ops.KeyValueTensorInitializer(keys, values),
146 lookup_ops.KeyValueTensorInitializer(
167 lookup_ops.KeyValueTensorInitializer(keys, values),
188 lookup_ops.KeyValueTensorInitializer(keys, values),
192 lookup_ops.KeyValueTensorInitializer(keys, values),
196 lookup_ops.KeyValueTensorInitializer(keys, values),
224 lookup_ops.KeyValueTensorInitializer(keys, values),
242 lookup_ops.KeyValueTensorInitializer(keys, values),
260 lookup_ops.KeyValueTensorInitializer(keys, values),
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.lookup.-key-value-tensor-initializer.pbtxt1 path: "tensorflow.lookup.KeyValueTensorInitializer"
3 is_instance: "<class \'tensorflow.python.ops.lookup_ops.KeyValueTensorInitializer\'>"
H A Dtensorflow.lookup.pbtxt4 name: "KeyValueTensorInitializer"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.lookup.-key-value-tensor-initializer.pbtxt1 path: "tensorflow.lookup.KeyValueTensorInitializer"
3 is_instance: "<class \'tensorflow.python.ops.lookup_ops.KeyValueTensorInitializer\'>"
H A Dtensorflow.lookup.pbtxt4 name: "KeyValueTensorInitializer"
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dhashtable.md93 initializer = tf.lookup.KeyValueTensorInitializer(string_values, int64_values)
109 initializer = tf.lookup.KeyValueTensorInitializer(string_values, int64_values)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
H A Dhash_table_v1.py63 table_initializer = tf.lookup.KeyValueTensorInitializer(
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/
H A Dstatic_hashtable.py45 initializer = tf.lookup.KeyValueTensorInitializer(key_tensor, value_tensor)
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/
H A Dstatic_hashtable.py41 initializer = tf.lookup.KeyValueTensorInitializer(key_tensor, value_tensor)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dlookup_ops.py516 class KeyValueTensorInitializer(TableInitializerBase): class
558 super(KeyValueTensorInitializer, self).__init__(self._keys.dtype,
1616 init = KeyValueTensorInitializer(
1780 init = KeyValueTensorInitializer(
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistributed_table_test.py80 return lookup_ops.KeyValueTensorInitializer(keys_tensor, vals_tensor)
331 self.initializer = lookup_ops.KeyValueTensorInitializer(
H A Dtpu_strategy_test.py280 initializer = lookup_ops.KeyValueTensorInitializer(
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Diterator_cluster_test.py127 lookup_ops.KeyValueTensorInitializer(keys, values),
H A Dtest_base.py342 return lookup_ops.KeyValueTensorInitializer(keys_tensor, vals_tensor)
H A Dmap_test.py402 lookup_ops.KeyValueTensorInitializer(keys, values), default_val)
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtest_util_test.py121 lookup_ops.KeyValueTensorInitializer(keys, values), default)
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dload_test.py1419 table1_initializer = lookup_ops.KeyValueTensorInitializer(keys, values)
2361 lookup_ops.KeyValueTensorInitializer(
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dlite_v2_test.py1540 table_initializer = tf.lookup.KeyValueTensorInitializer(