def insert(self, tree_ids, node_ids, logits): """Inserts values and returns the op.""" insert_op = lookup_ops.lookup_table_insert_v2( self._table_ref, self._example_ids, array_ops.concat([ array_ops.expand_dims( array_ops.bitcast(tree_ids, dtypes.float32), 1), array_ops.expand_dims( array_ops.bitcast(node_ids, dtypes.float32), 1), logits, ], axis=1, name='value_concat_for_cache_insert')) return insert_op
def insert(self, tree_ids, node_ids, logits): """Inserts values and returns the op.""" insert_op = lookup_ops.lookup_table_insert_v2( self._table_ref, self._example_ids, array_ops.concat( [ array_ops.expand_dims( array_ops.bitcast(tree_ids, dtypes.float32), 1), array_ops.expand_dims( array_ops.bitcast(node_ids, dtypes.float32), 1), logits, ], axis=1, name='value_concat_for_cache_insert')) return insert_op